Ubuntu20.04版本安装cuda和cudNN(宝宝级攻略)
前言
我在学习深度学习时,在Ubuntu系统下安装cuda和cudNN踩过了一些坑,浪费了很多的时间,现在想出一个宝宝级的攻略,希望能够帮助大家节约时间,规避一些毒教程的糟粕。
如果大家通过我的攻略安装成功,请在评论区打出“好用“希望能够帮助到更多人。
1.安装前的准备
这很重要:下载前请确认已经在Ubuntu系统下已经下载好了显卡驱动,如果没安装好请看这篇文章进行安装:NVIDIA显卡的Ubuntu20.04版本的驱动程序安装(宝宝级攻略)。
打开终端输入命令:
nvidia-smi
查询你的显卡对应的cuda版本信息,这里我对应的时12.4版本。
2.安装cuda
由于是外国网站没有“vip”可能会慢一点请耐心等待。
1.进入官网
https://developer.nvidia.com/cuda-toolkit-archive
一定要对应版本!
选择右图所示的配置。
2.打开终端执行一些操作:
注意:下面这些命令是根据你选择的配置网页下方自动生成的,请直接复制网页下方的链接,当然如果你和我的配置一样,请直接复制我的运行即可。
wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run
sudo sh cuda_12.4.0_550.54.14_linux.run
1.accept
2.取消勾选Driver,选择install。
注意:取消勾选前边的[]中没有X。
3.安装完显示如下:
3.修改环境变量
vim ~/.bashrc
没有vim请输入下列命令进行安装。
sudo apt-get install vim
打开bashrc后输入如下指令:
注意:这里的路径与上一幅图的路径相一致。
export PATH=/usr/local/cuda-11.4/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
输入:
source ~/.bashrc
nvcc -v
若找不到命令,请根据他的提示进行下载。
sudo apt install nvidia-cuda-toolkit
最后显示如下:
3.安装cudNN
1.进入官网
https://developer.nvidia.com/rdp/cudnn-archive
我是12.4所以选择了12.x的。最好选择新的版本,一般新版本会向下兼容。
2.提醒
这里下载cuDNN时,官网会强制让你去注册它的账户,这是不可避免的,请您按照它的要求注册一个账号即可。
3.下载并解压
打开文件进入如下界面:
输入如下指令:
sudo cp include/cudnn.h /usr/local/cuda-12.4/include
sudo cp lib/libcudnn* /usr/local/cuda-12.4/lib64
sudo chmod a+r /usr/local/cuda-12.4/include/cudnn.h /usr/local/cuda-12.4/lib64/libcudnn*
注意:对应自己的cuda版本。
4.结束语
希望我的宝宝级攻略能够真的帮助到了大家。如果真的对大家有帮助,请给我一键三连吧,您的支持是我前进的动力。