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

Question

在Ubuntu20.10上安装CUDA似乎有几种选择:它与20.10捆绑在一起，在NVIDIA官方页面上有各种安装程序，等等。

问:在Ubuntu 20.10上安装CUDA 11.X的推荐方法是什么?如何验证安装？

Answer 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。

测试数据自动化系统工具包安装/configuration

验证是否正确安装数据自动化系统的最佳方法之一是使用正式的“数据自动化系统-样本”。Ubuntu没有将它们打包成“NVIDIA-cuda-工具箱”的一部分，但是我们可以直接从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. 您可以通过运行nvidia-smi来查找您的GPU。我的是Quadro RTX 3000。
2. 接下来，谷歌你的GPU找出相应的计算架构。对于Quadro RTX 3000，它是“图灵”，版本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