![如何在 Ubuntu 20.10 上安装 CUDA 11 并验证安装](https://linux22.com/image/1069685/%E5%A6%82%E4%BD%95%E5%9C%A8%20Ubuntu%2020.10%20%E4%B8%8A%E5%AE%89%E8%A3%85%20CUDA%2011%20%E5%B9%B6%E9%AA%8C%E8%AF%81%E5%AE%89%E8%A3%85.png)
在 Ubuntu 20.10 上安装 CUDA 似乎有几种选择:它与 20.10 预先捆绑在一起,有各种安装程序NVIDIA 官方页面, ETC。
问题:在 Ubuntu 20.10 上安装 CUDA 11.X 的推荐方法是什么,以及如何安装核实安装?
答案1
安装 NVIDIA 驱动程序
这可能是可选步骤,但最好先删除可能先前安装的 NVIDIA 驱动程序:
sudo apt-get purge *nvidia*
sudo apt autoremove
接下来我们安装最新的驱动程序:
sudo apt install nvidia-driver-455
之后,我们需要重新开始计算机完成驱动程序安装。接下来我们可以验证驱动器是否安装成功:
nvidia-smi
其中应包含以下内容或类似内容:
NVIDIA-SMI 455.28 驱动程序版本:455.28
安装 CUDA 工具包
接下来我们可以安装 CUDA 工具包:
sudo apt install nvidia-cuda-toolkit
我们还需要设置 CUDA_PATH。添加此
export CUDA_PATH=/usr
.bashrc
在你的和运行的末尾
source ~/.bashrc
现在你的 CUDA 安装应该已完成,并且
nvidia-smi
应该表明您已经安装了 CUDA 11.1。
测试 CUDA 工具包安装/配置
验证 CUDA 是否正确安装的最佳方法之一是使用官方的“CUDA-sample”。Ubuntu 没有将它们打包为“nvidia-cuda-toolkit”的一部分,但我们可以直接从 NVIDIA 的 github 页面下载它们:
wget https://github.com/NVIDIA/cuda-samples/archive/v11.1.tar.gz
tar xvf v11.1.tar.gz
cd cuda-samples-11.1
无论出于什么原因,NVIDIA 没有选择包含现代构建系统(例如 cmake),而是提供了一个普通的旧 Makefile。如果仅运行“make”对您不起作用,请仔细阅读错误消息并查看是否未安装某些必需的依赖项。
为了稍微帮助构建过程,建议指定 GPU 的计算架构。
- 您可以通过运行来找到您的 GPU
nvidia-smi
。我的是 Quadro RTX 3000。 - 接下来,谷歌搜索您的 GPU,找出相应的计算架构。对于 Quadro RTX 3000,它是“turing”,版本 7.5。
- 运行 make 时指定体系结构版本,例如
make SMS="75"
如果编译成功,你可以尝试其中一个示例。例如:
./bin/x86_64/linux/release/immaTensorCoreGemm
您应该看到以下或类似的输出:
M: 4096 (16 x 256)
N: 4096 (16 x 256)
K: 4096 (16 x 256)
Preparing data for GPU...
Required shared memory size: 64 Kb
Computing... using high performance kernel compute_gemm_imma
Time: 6.030176 ms
TOPS: 22.79
答案2
由于到目前为止我找到的所有解释都不令人满意,下面是我提出的安装带有 cuda 的最新 nvidia 驱动程序 (465) 的步骤11.3
首先你必须卸载所有与 cuda 和 nvidia 相关的驱动程序和软件包
sudo apt-get purge nvidia-*
sudo apt-get purge cuda*
sudo apt autoremove
然后
reboot
然后(如果尚未完成)按照此处所述禁用 nouveau:
https://askubuntu.com/a/951892/634048然后reboot
再次
下载 cuda 并安装。
wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run
sudo sh cuda_11.3.1_465.19.01_linux.run
按照安装后说明进行操作Linux 版 CUDA 工具包安装指南。这涉及更新 PATH 和环境变量:
export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64\
${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
重要的如果您需要 32 位支持 - 有几个应用程序只能使用 32 位驱动程序运行(例如 steam)
选择退出安装用于 cuda 安装的 nvidia 驱动程序并从这里安装驱动程序: https://www.nvidia.com/download/driverResults.aspx/171980/en-us 还要检查驱动程序是否与您的型号兼容!(一般来说应该是这样的)
sudo sh 'NVIDIA-Linux-x86_64-465.19.01.run'
和选择 32 位支持当被问及
完毕
我没有测试过任何其他版本20.04
,但它应该适用18.04
于21.04