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