Ubuntu20.04版本安装cuda和cudNN(宝宝级攻略)

前言

我在学习深度学习时,在Ubuntu系统下安装cuda和cudNN踩过了一些坑,浪费了很多的时间,现在想出一个宝宝级的攻略,希望能够帮助大家节约时间,规避一些毒教程的糟粕。

如果大家通过我的攻略安装成功,请在评论区打出“好用“希望能够帮助到更多人。

1.安装前的准备

这很重要:下载前请确认已经在Ubuntu系统下已经下载好了显卡驱动,如果没安装好请看这篇文章进行安装:NVIDIA显卡的Ubuntu20.04版本的驱动程序安装(宝宝级攻略)。

打开终端输入命令:

nvidia-smi

查询你的显卡对应的cuda版本信息,这里我对应的时12.4版本。

693f44de5a1b40a0b5106bcb425f998f.png

2.安装cuda

由于是外国网站没有“vip”可能会慢一点请耐心等待。

1.进入官网
https://developer.nvidia.com/cuda-toolkit-archive

一定要对应版本!

3f3e4d65889b4c918e664fcf0be7b247.png178341ab60ad495796d1f4f7475b87ab.png

选择右图所示的配置。

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 56da347f6f7940bbacef608a8559dd99.png

2.取消勾选Driver,选择install。

注意:取消勾选前边的[]中没有X。

d3970442236f4910ab6d4b68d1bc08ac.png 3.安装完显示如下:f1cb3a2b14e6455b879d7236c01ac6d9.png

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

 若找不到命令,请根据他的提示进行下载。

f96f93b76fb140d8a83ef745472beac4.png

sudo apt install nvidia-cuda-toolkit

最后显示如下: 

 884ea58408d74a5e852c165f893116f4.png

3.安装cudNN

1.进入官网
https://developer.nvidia.com/rdp/cudnn-archive

我是12.4所以选择了12.x的。最好选择新的版本,一般新版本会向下兼容。

ffdca66e67b14721a08ff9b836401200.png

2.提醒

这里下载cuDNN时,官网会强制让你去注册它的账户,这是不可避免的,请您按照它的要求注册一个账号即可。

3.下载并解压

打开文件进入如下界面:

7317f6b26dbd456ba949ec3ee01368ac.png

输入如下指令:

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.结束语

希望我的宝宝级攻略能够真的帮助到了大家。如果真的对大家有帮助,请给我一键三连吧,您的支持是我前进的动力。

评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值