手把手教你在 Ubuntu16.04 安装 GPU 驱动 + CUDA9.0 + cuDNN7

如果你的电脑上安装了 ubuntu 16.04 系统，并且配备了一块 nvidia geforce gpu 显卡，那么不利用它来运行深度学习模型就太可惜了！尽管网上有许多相关教程，但质量参差不齐。本文将详细指导你如何在 ubuntu 16.04 上安装 gpu 显卡驱动、cuda 9.0 和 cudnn 7，值得一试！

注意：由于不同系统和 GPU 型号对应不同的 CUDA 和 cuDNN 版本，本文仅针对 Ubuntu 16.04、NVIDIA GeForce GPU、CUDA 9.0 和 cuDNN 7 进行说明。其它版本的安装过程类似，但细节上可能有所不同，请根据实际情况进行调整。

必要条件

要顺利完成安装，你的电脑需要满足以下两个条件：

1. 安装了 Ubuntu 16.04 操作系统。
2. 配备一块 NVIDIA GeForce GPU 显卡。

一、安装 GPU 显卡驱动

仅仅拥有一块 NVIDIA GPU 显卡还不够，你还需要安装相应的 GPU 显卡驱动。

1. 首先，访问 GeForce 官方网站：

   https://www.php.cn/link/a1141964749a018122bcc2d44abecb1f

   在手动搜索驱动程序部分，选择你的系统版本，确保选择的是“Linux 64-bit”。其他选项保持默认即可。点击开始搜索，网站会自动搜索与你的电脑匹配的 GPU 显卡驱动，搜索结果如图所示：

   

   如图所示，显示了可用的 GPU 显卡驱动程序。确定你需要的版本，直接点击下载。例如，我选择的版本是 418.43，下载后的文件名为 NVIDIA-Linux-x86_64-418.43.run。

2. 打开终端 Terminal，首先卸载系统中之前安装的 NVIDIA 显卡驱动，以防冲突。（如果之前没有安装，可以跳过这一步）

   $ sudo apt-get --purge remove nvidia-*

   登录后复制

3. 打开黑名单文件：

   $ sudo gedit /etc/modprobe.d/blacklist.conf

   登录后复制

   在文件末尾添加以下几行：

   blacklist vga16fb
   blacklist nouveau
   blacklist rivafb
   blacklist nvidiafb
   blacklist rivatv

   登录后复制

   然后在终端输入以下命令检查是否已禁用 nouveau 驱动：

   $ lsmod | grep nouveau

   登录后复制

   如果没有输出，可以继续下一步。如果有输出，则需要进一步禁用 nouveau 驱动。在终端输入：

   $ sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

   登录后复制

   直接写入以下内容：

   blacklist nouveau
   blacklist lbm-nouveau
   options nouveau modeset=0
   alias nouveau off
   alias lbm-nouveau off

   登录后复制

   保存并关闭文件，然后在终端输入以下命令验证是否已成功禁用 nouveau 驱动：

   $ sudo update-initramfs -u
   $ lsmod | grep nouveau
   # 如果还有输出，重启一次系统

   登录后复制

4. 正式安装驱动前，请记住你下载的 .run 文件的路径和完整文件名。按 Ctrl+Alt+F1 进入命令行模式，输入以下命令：

   $ sudo /etc/init.d/lightdm stop
   $ sudo init 3

   登录后复制

   进入 .run 文件的下载路径，使用 chmod 命令增加 .run 文件的执行权限：

   $ chmod a+x NVIDIA-Linux-x86_64-418.43.run
   $ sudo ./NVIDIA-Linux-x86_64-418.43.run -no-x-check -no-nouveau-check -no-opengl-files

   登录后复制

   安装程序将开始运行，安装完成后，输入以下命令检查驱动是否安装成功：

   $ nvidia-smi

   登录后复制

   如果看到类似下图的结果，证明驱动安装成功了：

   

   最后，回到图形界面：

   $ sudo /etc/init.d/lightdm start

   登录后复制

   这样，GPU 显卡驱动就正式安装成功了。

二、安装 CUDA 9.0

1. 首先，访问 CUDA 的下载网站：

   https://www.php.cn/link/3eebabe3234038516503146e19191ac5

   由于 CUDA 的最新版本已经更新到 10，我们需要安装 9.0 版本。点击“Legacy Releases”进入旧版本列表。

   

   在 CUDA 版本列表中，选择 9.0 版本。

   

   

   智写助手

   智写助手 写得更快，更聪明

   12

   查看详情

   选择对应的系统和其他信息，然后点击“Download”下载 runfile 文件。

   

2. 打开终端，进入 CUDA 9.0 的下载目录，输入以下命令：

   $ chmod a+x cuda_9.0.176_384.81_linux.run
   $ sudo ./cuda_9.0.176_384.81_linux.run

   登录后复制

   在安装过程中，会询问你一些配置信息，一般选择默认选项即可。需要注意的是，当询问是否安装 Graphics Driver for Linux-x86_64 时，必须选择“no”，否则会覆盖之前安装的驱动。

3. 安装完成后，设置环境变量。打开 .bashrc 文件，在终端输入：

   $ sudo gedit ~/.bashrc

   登录后复制

   在文件末尾添加以下内容：

   export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
   export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

   登录后复制

   保存并退出，然后在终端输入：

   $ source ~/.bashrc

   登录后复制

4. 验证 GPU 驱动版本：418.43

   $ cat /proc/driver/nvidia/version

   登录后复制

   

   验证 CUDA 版本：9.0

   $ nvcc -V

   登录后复制

   

5. 可能遇到的问题是没有安装 gcc 或 gcc 版本过高，需要手动降级，但这些问题通常容易解决，这里不再详细说明。

三、安装 cuDNN 7

1. 首先，访问 cuDNN 的下载网站：

   https://www.php.cn/link/4056b88955b3937af025deed0c8606c0

   下载之前需要先注册账号。在版本列表中，选择适合 CUDA 9.0 的版本，对应的是 Linux 系统。

   

2. 下载完成后，解压文件：

   $ tar -zxvf cudnn-9.0-linux-x64-v7.6.1.34.tgz

   登录后复制

   解压命令输出如下：

   cuda/include/cudnn.h
   cuda/NVIDIA_SLA_cuDNN_Support.txt
   cuda/lib64/libcudnn.so
   cuda/lib64/libcudnn.so.7
   cuda/lib64/libcudnn.so.7.6.1
   cuda/lib64/libcudnn_static.a

   登录后复制

   将相关文件复制到指定目录：

   $ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
   $ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

   登录后复制

3. 进入本地 CUDA 路径，创建软连接，终端输入：

   $ cd /usr/local/cuda/lib64
   $ sudo chmod +r libcudnn.so.7.6.1
   $ sudo ln -sf libcudnn.so.7.6.1 libcudnn.so.7
   $ sudo ln -sf libcudnn.so.7 libcudnn.so
   $ sudo ldconfig

   登录后复制

   注意，这里文件名和版本号应根据实际解压文件进行调整。

4. 在本地 CUDA 路径下进行检测：

   $ cd /usr/local/cuda/lib64
   $ ll

   登录后复制

   如果看到 cuDNN 对应的版本号，就证明安装成功了！

   lrwxrwxrwx 1 root root 13 3月 19 16:02 libcudnn.so -> libcudnn.so.7*
   lrwxrwxrwx 1 root root 17 3月 19 16:02 libcudnn.so.7 -> libcudnn.so.7.6.1*
   -rwxr-xr-x 1 root root 287641664 3月 19 16:00 libcudnn.so.7.6.1*

   登录后复制

大功告成！GPU 显卡驱动、CUDA 9.0 和 cuDNN 7 的环境配置已经完成。

参考文献：

https://www.php.cn/link/7f2bbe4bef9df4c8b49a5603e31b0022

以上就是手把手教你在 Ubuntu16.04 安装 GPU 驱动 + CUDA9.0 + cuDNN7的详细内容，更多请关注php中文网其它相关文章！