介绍

CUDA 工具包是英伟达开发的图形处理单元并行计算架构的扩展。 它还通过使用 GPU 的强大功能来增强计算机的性能。 程序员可以通过流行的编程语言使用它。 我们可以从 Ubuntu 存储库安装 CUDA 工具包。

以下是有关如何在 Ubuntu 20.04 上安装 CUDA 的指南。

从 Ubuntu 存储库安装 CUDA 工具包

步骤 1 – 通过 apt 命令从 Internet 更新软件包：

$ sudo apt update

输出：

第 2 步 – 现在我们开始通过 apt 命令安装 CUDA 工具包：

$ sudo apt install nvidia-cuda-toolkit

输出：

将 CUDA 添加到 PATH

安装后，我们必须在PATH中加入CUDA，让shell知道CUDA所在的位置。 让我们通过 vim 命令打开 .profile 文件：

$ sudo vim ~/.profile

然后将以下行添加到文件末尾并保存：

# set PATH for cuda 10.1 installation

if [ -d "/usr/local/cuda-10.1/bin/" ]; then

export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

fi

完成后，让我们重新启动计算机：

$ sudo reboot

检查您的 CUDA 版本

$ nvcc --version

输出：

结论

您已经详细了解了如何在 Ubuntu 20.04 上安装 CUDA。

谢谢阅读。