安装CUDA时，如何确定发行版、版本和体系结构参数？

Question

安装指南要求运行以下命令：

sudo dpkg -i cuda-repo-<distro>_<version>_<architecture>.deb

但我该如何填写distro、version和architecture呢？

Answer 1

如果您使用数据自动化系统下载网站并选择适当的选项(例如Linux，x86-64，Ubuntu，16.04，按顺序排列)，它将显示您应该运行的确切命令：

• 发行版是Ubuntu (16.04或14.04取决于您使用的是什么，请运行lsb_release -cd查找)。
• 该版本是基于您正在下载的CUDA版本。
• 有关体系结构，请参见如何找到我的个人电脑和Ubuntu的架构？ -使用uname -m。

Answer 2

CUDA下载网站只反映了CUDA的最新版本。您还应该检查以下内容：

1. 推荐硬件驱动程序。对于我来说，在Ubuntu20.04上，我的RTX 3090显卡应该使用470版本的驱动程序。参见发布说明，表3中对应的最大CUDA版本。就我而言，是CUDA 11.4.x。
2. GCC的要求，您的数据自动化系统版本。在Ubuntu20.04上，默认的GCC是9.x，所以我的最大CUDA版本是11.4.0 (我不想安装一个更新的GCC)。
3. 如果您正在使用依赖于一个特定版本的CUDA的其他包，也可以检查这些包(例如，当前的最新Pytorch是用CUDA 11.3编译的)。
4. 注意您选择的数据自动化系统的驱动程序版本:对于11.3.1，驱动程序版本是465.19.01。

下载网站告诉我使用cuda-repo-ubuntu2004-11-6-local_11.6.0-510.39.01-1_amd64.deb。我将相应地修改这些版本：

Distro:       ubuntu2004-11-6-local -> ubuntu2004-11-3-local
Version:      11.6.0-510.39.01-1    -> 11.3.1-465.19.01-1 (<CUDA version>-<driver version>)
Architecture: amd64 (no change)

因此，我将下载站点的输出更改为：

wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda-repo-ubuntu2004-11-6-local_11.6.0-510.39.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-6-local_11.6.0-510.39.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-6-local/7fa2af80.pub

至

wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda-repo-ubuntu2004-11-3-local_11.3.1-465.19.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-3-local_11.3.1-465.19.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-3-local/7fa2af80.pub

如果您安装了错误的版本(例如11.6)并想要撤消，您可以使用

sudo apt remove cuda
sudo dpkg remove cuda
sudo apt-key remove /var/cuda-repo-ubuntu2004-11-6-local/7fa2af80.pub
sudo rm /etc/apt/sources.list.d/cuda-ubuntu2004-11-6-local.list
sudo rm -r /usr/local/cuda-11.6 
sudo rm /etc/alternatives/cuda /etc/alternatives/cuda-11