Ubuntu 18.04 安装 NVIDIA+ CDUDA +CUDNN

NVIDIA显卡驱动安装

显卡驱动+cuda+cudnn不同版本 跳转链接

使用 Ubuntu 软件仓库中的稳定版本安装

在终端输入：

ubuntu-drivers devices

可以看到如下界面：（类似图片，显卡Geforce 1060 现在支持 460 和470系列）

安装步骤

接着需要先安装一些 NVIDIA 显卡依赖的软件，在终端依次执行如下命令：

sudo dpkg --add-architecture i386
sudo apt update
sudo apt install build-essential libc6:i386

Ubuntu 系统默认安装好是使用的一个开源的驱动：nouveau，我们要安装官方的驱动需要先禁用这个开源驱动，方法如下，依次执行：

sudo bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nvidia-nouveau.conf"

sudo bash -c "echo options nouveau modeset=0 >> /etc/modprobe.d/blacklist-nvidia-nouveau.conf"

执行完上面两条指令后，我们使用如下命令看看是否成功禁用了开源驱动：
cat /etc/modprobe.d/blacklist-nvidia-nouveau.conf。如果和下面一样，表示成功了。

然后这个时候我们需要先重启一下系统

找到下载位置，运行.run

执行 .run 文件

sudo sh NVIDIA-linux-XXXXXX.run

如果出现安装错误

An NVIDIA kernel module ‘nvidia-drm’ appears to already be loaded in your kernel…

那么就输入这一行代码

sudo systemctl isolate multi-user.target

sudo modprobe -r nvidia-drm

重启电脑后，电脑分辨率下降，不要慌，我们继续设置，运行.run

运行结束后，重启电脑，分辨率回来了
运行代码进行测试，看显卡驱动是否安装成功

nvidia-smi

cuda

然后，

sudo vim ~/.bashrc

我们在文件最后一行添加：

export PATH="/usr/local/cuda-11.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH"

最后，使刚才的配置生效。

source ~/.bashrc

nvcc -version

卸载 nvidia-cuda-toolkit

sudo apt-get autoremove nvidia-cuda-toolkit

安装cuDNN

然后解压，并进入到相应目录，运行以下命令：

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

以配置cuDNN环境。

通过

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

可以查看cuDNN版本。