Linux 安装 CUDA Toolkit 笔记

贡献者： addis

　　 2024-1-15 更新：Ubuntu 22.04 无法安装 Nvidia 驱动，因为编译 Linux 内核使用的 gcc 11.3.0 不可能正常安装！ CUDA Toolkit 只有 20.04 和 22.04 两个选项，所以系统只能用 20.04！

　　 如果你装机放了一张 Nvidia 显卡，那么 Ubuntu 在安装时会自动提供一个 Nouveau 开源驱动。

　　 笔者使用 Ubuntu 20.04 和 GTX 1080 Ti 显卡进行测试。

1. 安装 CUDA 和驱动

　　 注意：不要改变系统默认的 gcc symlink 的版本！ 因为需要 gcc 编译 kernel module，必须使用默认版本。如果已经改变了，就卸载 gcc 和 g++，再重装一次就行。

　　 直接安装 cuda, 是包括显卡驱动的! (远程的话就用 Xwindow 打开 firefox) 网址。直接看 3.6 Ubuntu, 按照步骤来。

• 点击 pre-installation actions, 下载适合的 deb 安装包即可。网址
• sudo dpkg -i cuda-repo-<distro>_<version>_<architecture>.deb 最后一项是安装包的文件名
• sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
• sudo apt-get update
• sudo apt-get install cuda
• 按照 post-installation action 的说明来

　　 post-installation action

• (put this command to ~/.bashrc)
• export PATH=/usr/local/cuda-9.2/bin${PATH:+:${PATH}}
• (注意! cuda-9.2 以后可能会变，具体数字要 cd 到 local 文件夹中查看! 如果不执行这个，nvcc 命令就找不到。执行后，nvcc 和 nvprof 应该都能正常工作了。 剩下的步骤暂时还没有进行。

2. 只安装驱动

• 首先更新驱动，先删除以前驱动 sudo apt-get purge nvidia*
• 然后安装驱动 sudo add-apt-repository ppa:graphics-drivers，sudo apt-get update，sudo apt-get install nvidia-390（或者新的版本）
• 最后那三个数字是驱动的版本号，上这个网页可以找到某个显卡的最新驱动版本号。网址
• 重新启动服务器！
• 驱动安装完成后，nvidia-smi 命令会显示显卡的信息。