初涉Linux炼丹术：基于Ubuntu 16.04的NVIDIA显卡驱动及CUDA、cuDNN安装全攻略
2023-12-11 20:13:41
在 Ubuntu 16.04 中驭驾 NVIDIA 显卡驱动、CUDA 和 cuDNN
踏入深度学习和人工智能的浩瀚领域，NVIDIA 显卡驱动、CUDA 和 cuDNN 是必不可少的利器。在 Ubuntu 16.04 系统中驾驭这些组件，助你开启探索之旅。
驱动禁术：禁用 Nouveau
首先，施展驱动禁术——禁用 Nouveau，它是 Ubuntu 系统中默认的开源显卡驱动。它与 NVIDIA 专有驱动共存时，会导致冲突，让你的显卡罢工。
要禁用 Nouveau，打开终端并施法：
sudo nano /etc/modprobe.d/blacklist-nouveau.conf
在弹出的魔法书中，添加两行神秘咒语：
blacklist nouveau
blacklist lbm-nouveau
保存法术，使用这句咒语更新内核：
sudo update-initramfs -u
重启系统，让法术生效。
软件秘笈：安装 NVIDIA 显卡驱动
禁用 Nouveau 后，就可以安装 NVIDIA 显卡驱动了。前往 NVIDIA 官网，下载与你的显卡型号和操作系统版本相匹配的驱动程序。
下载完成后，使用这句咒语安装驱动：
sudo sh NVIDIA-Linux-x86_64-XXX.XX.run
其中，XXX.XX 是你下载的驱动版本号。按照提示完成安装。
环境配置：安装 CUDA 和 cuDNN
NVIDIA 显卡驱动安装完毕，接下来安装 CUDA 和 cuDNN，为深度学习和人工智能计算赋能。
安装 CUDA
首先，从 NVIDIA 官网下载与你的显卡型号和操作系统版本相匹配的 CUDA 软件包。下载完成后，用这句咒语安装 CUDA：
sudo sh cuda_xxx.xx_linux.run
其中，xxx.xx 是你下载的 CUDA 版本号。按照提示完成安装。
安装 cuDNN
下载 cuDNN 软件包，解压到以下目录：
/usr/local/cuda/lib64
最后，用这句咒语创建符号链接：
sudo ln -s /usr/local/cuda/lib64/libcudnn* /usr/local/cuda/lib64/
扫清障碍：解决 Ubuntu 循环登录问题
安装完 NVIDIA 显卡驱动后，你可能遭遇 Ubuntu 循环登录问题。这是因为 Xorg 无法识别你的显卡。要解决这个问题，编辑 Xorg 配置文件：
sudo nano /etc/X11/xorg.conf
在文件中找到 Device 部分，添加以下咒语：
Section "Device"
Identifier "NVIDIA Card"
Driver "nvidia"
VendorName "NVIDIA Corporation"
EndSection
保存并关闭文件。
结语：炼丹术大成
恭喜你！你已成功在 Ubuntu 16.04 系统中安装了 NVIDIA 显卡驱动、CUDA 和 cuDNN，并解决了 Ubuntu 循环登录问题。现在，你可以畅游深度学习和人工智能的世界了。
常见问题解答
- 为什么需要禁用 Nouveau？
Nouveau 与 NVIDIA 专有驱动冲突，导致系统不稳定。禁用 Nouveau 可确保显卡顺利运行。
- 在哪里下载 NVIDIA 显卡驱动？
前往 NVIDIA 官网下载与你的显卡型号和操作系统版本相匹配的驱动程序。
- 如何安装 CUDA 和 cuDNN？
从 NVIDIA 官网下载 CUDA 软件包，然后使用提供的脚本安装。对于 cuDNN，下载并解压到指定目录，并创建符号链接。
- 如果遇到 Ubuntu 循环登录问题怎么办？
编辑 Xorg 配置文件，添加相应的代码，以识别 NVIDIA 显卡。
- 安装完成后，如何确认是否成功？
打开终端，输入 nvidia-smi。如果显示你的显卡信息，则表明安装成功。
