cuda需要Nvidia显卡或计算卡,AMD或intel显卡不行(但是也有套他们的标准)
就算是亮机卡也可使用,比如GT710这种
建议使用ubuntu来装,因为cuda就是在这个平台上开发的,当然别的linux系统也行
以下操作在ubuntu server 2204,如果没有安装ubuntu系统,可以参考 Ubuntu server安装图解
注意!请根据需要安装对应版本的cuda!不同版本安装大同小异
主要思路:安装N卡(硬件),安装依赖、安装N卡驱动,安装nvcc、安装cuda
另外,你可能还需要pytorch等常用的组件,在选择版本前务必根据你自己的需要选取,部分组件对操作系统也有要求,为了少做重复劳动,先把所有需要的组件版本对应好,然后再一个个安装。
cuda官方安装文档
lspci | grep -i nvidia
gcc --version
如果没有,建议安装这个c的一大堆套件集合,一劳永逸
apt-get install build-essential
apt-get install linux-headers-$(uname -r)
推荐使用,这样全家桶就装好了:N卡驱动+cuda+nvcc
注意:N卡驱动的版本和cuda是强对应的,所以这种方法不需要先装驱动,也省去了查找兼容性的问题
官方地址: cuda toolkit
这样选择,下面会出来安装的命令,直接复制用就行
这里装的是12.1版本,在上面的官方地址可以选取自己需要的版本,方法都是类似的
下面根据命令一步步操作:
mkdir /usr/local/my_cuda && cd /usr/local/my_cuda
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
cp /var/cuda-repo-ubuntu2204-12-1-local/cuda-FDA658E8-keyring.gpg /usr/share/keyrings/
更新
apt-get update
安装cuda,这一步时间较长,耐心等待
apt-get -y install cuda
安装完成后重启电脑,否则可能会出各种问题
nvcc -V
显示如下:
nvidia-smi
如果这两步都出现问题,第四节处理
查找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
比如如下图提示,如果错误,重启即可,因为上面已经装过了,很多问题重启就能解决
再次使用nvidia-smi可以看到类似下面的信息(左上角是N卡驱动版本,右上角是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