如何在 Ubuntu 20.04 LTS 上安装 CUDA

2023年 7月 21日 28.4k 0

CUDA 代表计算统一设备架构。CUDA 包括 CUDA 工具包(编译器、配置文件和调试器)、软件驱动程序和 CUDA SDK。软件层可以直接访问 GPU 的虚拟指令集和并行计算元素。对于深度学习研究和框架开发人员,使用 cuDNN 进行高性能 GPU 加速。Nvidia-logo

在 Ubuntu 20.04 LTS Focal Fossa 上安装 CUDA

步骤 1. 首先,通过apt在终端中运行以下命令确保所有系统包都是最新的。

sudo apt update
sudo apt upgrade

步骤 2. 在 Ubuntu 20.04 上安装 CUDA。

默认情况下,CUDA 包在 Ubuntu 20.04 基础存储库上不可用。现在我们使用以下命令将 CUDA 存储库 pin 添加到您的系统:

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin

接下来,我们将引脚移动到目录并将其重命名为:/etc/preferences.dcuda-repository-pin-600

sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600

之后,使用以下命令下载 CUDA Ubuntu 文件:

wget https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

然后,使用dpkg命令执行文件:

sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

最后,软件包列出并安装CUDA软件包:

sudo apt update
sudo apt install cuda

步骤 3. 安装 CuDNN。

现在我们从官方页面下载安装的CuDNN。注册 NVIDIA Developer Program 后,您可以下载最新版本的 cuDNN:

tar -xzvf cudnn-10.2-linux-x64-v8.2.4.15.tgz

接下来,将以下文件复制到 CUDA 目录:

sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

感谢您使用本教程在 Ubuntu 20.04 LTS Focal Fossa 系统上安装 CUDA NVIDIA。如需其他帮助或有用信息,我们建议您查看NVIDIA 官方网站。

相关文章

服务器端口转发,带你了解服务器端口转发
服务器开放端口,服务器开放端口的步骤
产品推荐:7月受欢迎AI容器镜像来了,有Qwen系列大模型镜像
如何使用 WinGet 下载 Microsoft Store 应用
百度搜索:蓝易云 – 熟悉ubuntu apt-get命令详解
百度搜索:蓝易云 – 域名解析成功但ping不通解决方案

发布评论