cuda 安装和卸载 (Ubuntu 2204)

cuda需要Nvidia显卡或计算卡,AMD或intel显卡不行(但是也有套他们的标准)
就算是亮机卡也可使用,比如GT710这种
建议使用ubuntu来装,因为cuda就是在这个平台上开发的,当然别的linux系统也行
以下操作在ubuntu server 2204,如果没有安装ubuntu系统,可以参考 Ubuntu server安装图解


注意!请根据需要安装对应版本的cuda!不同版本安装大同小异
主要思路:安装N卡(硬件),安装依赖、安装N卡驱动,安装nvcc、安装cuda
另外,你可能还需要pytorch等常用的组件,在选择版本前务必根据你自己的需要选取,部分组件对操作系统也有要求,为了少做重复劳动,先把所有需要的组件版本对应好,然后再一个个安装。
cuda官方安装文档

一、检查硬件软件环境

1.确定系统识别了N卡

lspci | grep -i nvidia

显示类似如下信息(我这里是rtx2060 12G):
在这里插入图片描述

2.检查gcc编译器

gcc --version

如果没有,建议安装这个c的一大堆套件集合,一劳永逸

apt-get install build-essential

3.检查相关配套程序是否安装

apt-get install linux-headers-$(uname -r)

二、使用cuda Toolkit安装

推荐使用,这样全家桶就装好了:N卡驱动+cuda+nvcc
注意:N卡驱动的版本和cuda是强对应的,所以这种方法不需要先装驱动,也省去了查找兼容性的问题
官方地址: cuda toolkit

这样选择,下面会出来安装的命令,直接复制用就行
这里装的是12.1版本,在上面的官方地址可以选取自己需要的版本,方法都是类似的
cuda 安装和卸载 (Ubuntu 2204)_第1张图片
cuda 安装和卸载 (Ubuntu 2204)_第2张图片
下面根据命令一步步操作:

1.先切换到程序下载目录

mkdir /usr/local/my_cuda && cd  /usr/local/my_cuda

2.安装操作

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin

移动配置文件

mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600

下载安装包

wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda-repo-ubuntu2204-12-1-local_12.1.0-530.30.02-1_amd64.deb

安装

dpkg -i cuda-repo-ubuntu2204-12-1-local_12.1.0-530.30.02-1_amd64.deb

安装key,不要用上面图的,上一步安装之后会提示
在这里插入图片描述

cp /var/cuda-repo-ubuntu2204-12-1-local/cuda-FDA658E8-keyring.gpg /usr/share/keyrings/

更新

apt-get update

安装cuda,这一步时间较长,耐心等待

apt-get -y install cuda

安装完成后重启电脑,否则可能会出各种问题

三、测试

1.测试nvcc(cuda编译器)

nvcc -V

显示如下:

2.测试nvidia-smi

nvidia-smi

如果这两步都出现问题,第四节处理

四、问题处理

1.nvcc 显示没有

查找nvcc

find / -name "nvcc"

比如出现下面目录
在这里插入图片描述

vi ~/.bashrc

最后的部分添加上

export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib
export PATH=$PATH:/usr/local/cuda-12.1/bin

保存后,刷新配置文件

source ~/.bashrc

再次使用命令

nvcc -V

见到如下的,是不是很亲切?
cuda 安装和卸载 (Ubuntu 2204)_第3张图片

2.nvidia-smi错误

比如如下图提示,如果错误,重启即可,因为上面已经装过了,很多问题重启就能解决
在这里插入图片描述再次使用nvidia-smi可以看到类似下面的信息(左上角是N卡驱动版本,右上角是cuda版本)

cuda 安装和卸载 (Ubuntu 2204)_第4张图片

五、cuda卸载

如果需要换不同的版本,建议多版本并存,这里先不涉及,如果要彻底卸载,根据以下操作

准备删除cuda

apt-get remove cuda

自动卸载

sudo apt autoremove 

删除其它的cuda

sudo apt autoremove cuda*

删除下载安装包(也可以不删除)

rm /usr/local/my_cuda/cuda-repo-ubuntu2204-12-1-local_12.1.0-530.30.02-1_amd64.deb

查找包相关

dpkg -l |grep cuda

删除相关的包,否则再安装别的版本会失败

sudo dpkg -P xxxx

你可能感兴趣的:(ai,ai)