Ubuntu中cuda配置

驱动安装

官网

查看是否安装成功

1
nvidia-smi

使用此指令可以查看,当前驱动支持的最高版本的驱动。

若安装成功,则会显示

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
sys120-1@sys120-1:~$ nvidia-smi
Mon Oct 27 09:09:11 2025
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 530.30.02 Driver Version: 530.30.02 CUDA Version: 12.1 |
|-----------------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|=========================================+======================+======================|
| 0 NVIDIA TITAN Xp On | 00000000:17:00.0 Off | N/A |
| 23% 32C P8 8W / 250W| 11MiB / 12288MiB | 0% Default |
| | | N/A |
+-----------------------------------------+----------------------+----------------------+
| 1 NVIDIA TITAN Xp On | 00000000:65:00.0 Off | N/A |
| 23% 36C P8 13W / 250W| 363MiB / 12288MiB | 2% Default |
| | | N/A |
+-----------------------------------------+----------------------+----------------------+

+---------------------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=======================================================================================|
| 0 N/A N/A 1202 G /usr/lib/xorg/Xorg 4MiB |
| 0 N/A N/A 28248 G /usr/lib/xorg/Xorg 4MiB |
| 1 N/A N/A 1202 G /usr/lib/xorg/Xorg 17MiB |
| 1 N/A N/A 28248 G /usr/lib/xorg/Xorg 242MiB |
| 1 N/A N/A 28378 G /usr/bin/gnome-shell 30MiB |
| 1 N/A N/A 28588 G ...in/bin/sunloginclient --cmd=autorun 17MiB |
| 1 N/A N/A 28668 G ...) Chrome/58.0.3029.81 Safari/537.36 3MiB |
| 1 N/A N/A 28731 G ...en=BBF5301FE07A582EABF97C6A2603FF90 15MiB |
| 1 N/A N/A 31739 G /opt/v2rayN/v2rayN 16MiB |
+---------------------------------------------------------------------------------------+

CUDA安装

在Linux中,可以同时存在不同版本cuda的,在使用时,只需要修改配置文件即可。

CUDA安装

根据操作系统选择安装

在官网中选择需要的cuda版本进行安装。

CUDA11.8.0安装示例

1
2
3
4
5
6
7
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-ubuntu2004-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-11-8

第7行,官网中使用sudo apt-get -y install cuda,而我们要在末尾加上指定的版本号,如sudo apt-get -y install cuda-11-8,否在安装的不是指定的版本,而是最新的。

修改配置文件

执行nano ~/.bashrc

在文件最后添加下面内容即可。

1
2
3
4
export CUDA_HOME=/usr/local/cuda-11.8
export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
export CUDA_ROOT=$CUDA_HOME

根据需要使用的cuda版本,进行修改内容

修改完成后,激活配置文件。

1
source ~/.bashrc

查看版本

查看所有安装版本

使用``查看所有安装完成的cuda版本。

1
ls -l /usr/local | grep cuda

查看当前使用版本

使用nvcc -V查看当前cuda的版本。

1
nvcc -V

成功安装CUDA11.8则显示

1
2
3
4
5
6
sys120-1@sys120-1:~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:33:58_PDT_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0

卸载cuda

卸载所有CUDA包

1
2
3
sudo apt-get --purge remove "*cuda*" "*nvidia*"
sudo apt-get autoremove
sudo apt-get autoclean

验证是否卸载完成

方法1

查找 nvcc 可执行文件的安装路径,显示当前环境中使用的 CUDA 编译器位置。

1
which nvcc

方法2

列出 /usr/local/ 目录下所有包含 “cuda” 的文件和文件夹,显示系统中安装的所有 CUDA 相关版本。

1
ls /usr/local/ | grep cuda