☰

如何在 Ubuntu 20.10 上安装 CUDA 11 并验证安装

tag-icon • tag-icon drivers nvidia cuda 20.10
如何在 Ubuntu 20.10 上安装 CUDA 11 并验证安装

在 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 的计算架构。

  1. 您可以通过运行来找到您的 GPU nvidia-smi。我的是 Quadro RTX 3000。
  2. 接下来，谷歌搜索您的 GPU，找出相应的计算架构。对于 Quadro RTX 3000，它是“turing”，版本 7.5。
  3. 运行 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

相关内容