ubuntu20.04安装12.2版本的cuda

访问如下网址

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=runfile_local
ubuntu20.04安装12.2版本的cuda_第1张图片

下载推荐的cuda并安装

复制箭头所指处的命令到命令行

wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run 

必须先关图形界面

如果不关,运行下面安装的命令之后会报错,装不上

systemctl isolate multi-user.target

关闭图形界面之后会重启,输入如下命令

 modprobe -r nvidia-drm

安装驱动

sudo sh cuda_12.2.0_535.54.03_linux.run

稍等一会
在弹出的安装界面中选“continue”,如果选了会跳出安装,就说明安装失败,给了失败日志的路径,自己查看原因,一般是gcc版本问题,降级方法如下:

sudo apt-get install gcc-7 g++-7
 
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 1
 
sudo update-alternatives --display gcc     //查看默认版本
 
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 9
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 1
 
sudo update-alternatives --display g++

如果没问题继续安装:
ubuntu20.04安装12.2版本的cuda_第2张图片
cuda安装包是自带显卡驱动的,所以这一步按空格去掉安装显卡驱动的选项,然后选择install
ubuntu20.04安装12.2版本的cuda_第3张图片

开启图形界面

sudo systemctl start graphical.target

配置CUDA的环境变量

成功后需要配置一下环境变量,这步很关键。一开始我忽略了这步,一直以为CUDA安装好了,但是TF运行的时候怎么都找不到GPU。

sudo gedit ~/.bashrc
然后查看自己的/usr/local中的cuda版本,我的是cuda-12.2

ubuntu20.04安装12.2版本的cuda_第4张图片

最后一行增加

export PATH=/usr/local/cuda-12.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

特别注意文件夹路径,有时候文件夹直接是/usr/local/cuda,有时候是别的版本号,要确保文件夹存在。

source ~/.bashrc

检查是否安装成功

注意:V是大写:
nvcc -V
ubuntu20.04安装12.2版本的cuda_第5张图片

卸载cuda

先进入安装目录下的bin文件夹中
cd /usr/local/cuda-12.2/bin

然后勾选我们要卸载的12.2版本,选择done
ubuntu20.04安装12.2版本的cuda_第6张图片
卸载成功
在这里插入图片描述

参考文献

https://blog.csdn.net/weixin_44857882/article/details/124108458
https://blog.csdn.net/qq_16792139/article/details/129730718

11.3版本cuda

https://developer.nvidia.com/cuda-11-3-1-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=runfile_local

必须先关图形界面

如果不关,运行下面安装的命令之后会报错,装不上

systemctl isolate multi-user.target

关闭图形界面之后会重启,输入如下命令

 modprobe -r nvidia-drm

安装驱动

sudo sh cuda_11.3.1_465.19.01_linux.run

开启图形界面

sudo systemctl start graphical.target

你可能感兴趣的:(ubuntu,linux,运维,深度学习,服务器)