Ubuntu CUDA 环境配置


典型 CUDA 环境配置

在 Ubuntu 20.04.6 LTS Desktop 上配置指定版本 CUDA 的实验环境的步骤如下:

  • 进入官网,找到下载界面
    图1
  • 选择 “Archive of Previous CUDA Releases”
    图2
  • 选择对应机器配置的安装包
    图2

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

Author: Yixiang Zhang
Reprint policy: All articles in this blog are used except for special statements CC BY 4.0 reprint policy. If reproduced, please indicate source Yixiang Zhang !
评论
  TOC