Ubuntu中cuda配置

驱动安装

官网

text

查看是否安装成功

1
nvidia-smi

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

卸载cuda

卸载所有CUDA包

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

验证是否卸载完成

1
which nvcc

ls /usr/local/ | grep cuda

安装

Cuda
Cuda
Cuda
源码

选择cuda版本

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

修改配置文件

执行nano ~/.bashrc。

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

1
2
export PATH=/usr/local/cuda-11.7/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64:$LD_LIBRARY_PATH

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

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

1
source ~/.bashrc

查看版本

查看所有安装版本

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

1
ls -l /usr/local | grep cuda

查看当前使用版本

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

1
nvcc -V