典型 CUDA 环境配置
在 Ubuntu 20.04.6 LTS Desktop 上配置指定版本 CUDA 的实验环境的步骤如下:
- 进入官网,找到下载界面
- 选择 “Archive of Previous CUDA Releases”
- 选择对应机器配置的安装包
Linux 的配置和安装要特殊一些。需要执行以下命令
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pinsudo
mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda-repo-ubuntu2004-11-6-local_11.6.2-510.47.03-1_amd64.debsudo
dpkg -i cuda-repo-ubuntu2004-11-6-local_11.6.2-510.47.03-1_amd64.debsudo
apt-key add /var/cuda-repo-ubuntu2004-11-6-local/7fa2af80.pubsudo apt-get updatesudo
apt-get -y install cuda
其含义如下:
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pinsudo
这条命令使用 wget
来下载文件。
mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
这里使用 mv
命令将文件移动到指定文件夹下。
wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda-repo-ubuntu2004-11-6-local_11.6.2-510.47.03-1_amd64.debsudo
作用同第一个 wget
。
dpkg -i cuda-repo-ubuntu2004-11-6-local_11.6.2-510.47.03-1_amd64.debsudo
dpkg
即 package manager for Debian ,是 Debian 和基于 Debian 的系统中一个主要的包管理工具,可以用来安装、构建、卸载、管理 deb 格式的软件包。
安装软件
使用 dpkg
命令安装软件时,可以使用 -i 选项并指定 deb 安装包的路径。和 Ubuntu 下的另一个包管理工具 apt-get
(Advanced Package Tool)有所不同。
apt-get
命令并不直接操作 deb 安装包文件,而是从 /etc/apt/sources.list 配置文件中定义的软件镜像源里下载软件包并安装,使用时也只需指定软件的名称(或者也可以附加上版本号)。
apt-get 安装软件
apt-get install
dpkg 命令安装软件:
dpkg -i
因此,dpkg
主要是用来安装已经下载到本地的 deb 软件包,或者对已经安装好的软件进行管理。而 apt-get
可以直接从远程的软件仓库里下载安装软件。
apt-key add /var/cuda-repo-ubuntu2004-11-6-local/7fa2af80.pubsudo
为 CUDA 添加 GPG key,用于验权。
apt-get updatesudo
不再赘述。
apt-get -y install cuda
不再赘述。
总的来说,这一系列代码完成了 CUDA 环境的配置,是 Linux 系统下典型的包安装实例,具有一定参考价值。
WSL CUDA 环境配置
安装旧版环境,尽量不要用 deb 包管理,直接用 runfile、tar,血的教训!!!
命令类似(来自官网):
sudo apt install nvidia-utils-550
sudo apt install nvidia-driver-550
wget https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run
sudo sh cuda_12.4.1_550.54.15_linux.run
cuda_12.4.1_550.54.15_linux.run // 补充 driver
sudo sh cuda_12.4.1_550.54.15_linux.run --silent --driver