0


零基础也能Linux Ubuntu20.04版本安装CUDA和cuDNN,我教你!

前言

用于深度学习的环境配置过程:新Ubuntu系统->安装显卡驱动->安装CUDA->安装cuDNN。

首先打开终端就是下面的黑色小框。

在桌面可以通过Ctrl+Alt+T打开,在文件夹可以通过鼠标右键选项里打开。

输入指令:

nvidia-smi #检查是否已经安装显卡驱动

如果显示上述样例就是已经安装好了。(ps:未安装的话去翻一下别人的文章,类似Ubuntu20.04安装Nvidia显卡驱动教程-CSDN博客

**下载 **

CUDA下载

下载官网:NVIDIA Developer

进入官网后点击platforms就可以找到CUDA和cuDNN的下载链接了。

接下来一步步点击即可。

到这里就可以下载各个版本的CUDA了。我这里以CUDA11.7.1为例。

配置:


补充说明:

1.至于为什么选择x86_64,你可以在终端输入下列代码

sudo uname -srnp #来确定Architecture

注:这里是以管理员身份运行(sudo),所以需要输入用户的密码。

这张图是后面补的,(base)在安装完anaconda后才会出现,这里不用管他。

2.选择runfile的原因:run文件自带更多的依赖包,更方便安装。


下面两行代码分为两部分,下载和安装。

把这个命令复制下来在小黑框(终端)里。

下载完即可。

cuDNN下载

因为我安装的是CUDA-11.7版本所以选择更多版本里的⬇(ps:这里需要注册登录邮箱下载。

点击就可以下载了

下载完成是默认在home/downloads文件夹的(不是的话最好移到一起吧,这样方便)。

安装

在Downloads文件夹下右键打开终端,这样的目的是为了cd当前路径到Downloads文件夹。

CUDA安装

复制下面的链接

注:这里是以管理员身份运行(sudo),所以需要输入用户的密码。

** ➡ **

通过↑↓键选中【x】Driver并按Enter键取消Driver的【x】。

安装成功。

安装好CUDA后,需要添加CUDA的环境变量。

输入下列指令打开bashrc文件并在里面编辑。

sudo vi ~/.bashrc # vi可以用vim和gedit替换,这些都是文本编译器,只不过vi是linux自带的。

打开如下


科普:vi文本编译器中:进入时默认是命令行模式就是只能选择在哪个地方开始修改,按【i】键后从当前光标位置进入插入模式(可编辑,不可移动上下键连鼠标滑轮都不能动!),或者按【o】键就可以自动换行再进入插入模式。


然后我们在底端加入路径:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.7/lib64
export PATH=$PATH:/usr/local/cuda-11.7/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.7
# 这三个代码的cuda-11.7都需要根据你安装的cuda版本替换
# 例如
# export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64
# export PATH=$PATH:/usr/local/cuda-11.1/bin
# export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.1

可以通过cd/usr/local再ls查看对应cuda-版本。

输入完毕如下:

按Esc返回到命令行模式,输入:wq Enter保存即可。


如果你像我一样手残容易打错的话,那就多按几次Esc返回到命令行模式,再输入:q! Enter即可。这样就是不保存且强制退出bashrc文件,再重复步骤把它重新打开即可重新编写。


验证安装完成

这里需要刷新终端配置!!!➡用 source ~/.bashrc或者关闭再打开终端都可。不然测不出来的。

nvcc-V #输入验证代码到终端

成功截图。

到这里CUDA就完成安装啦!

** cuDNN安装**

以下代码都是在Downloads文件夹下打开的终端运行。

1.解压刚刚下载好的文件

tar -xf cudnn-linux-x86_64-8.9.3.28_cuda11-archive.tar.xz 
# tar -xf 后面复制自己的cudnn压缩包文件名就行

2.复制文件到CUDA安装目录

sudo cp cudnn-linux-x86_64-8.9.3.28_cuda11-archive/include/* /usr/local/cuda-11.7/include
sudo cp cudnn-linux-x86_64-8.9.3.28_cuda11-archive/lib/libcudnn* /usr/local/cuda-11.7/lib64
# 依次输入这两行代码到终端即可。这里的cuda-版本和上面一样按自己的来。
# cudnn-linux-x86_64-8.9.3.28_cuda11-archive.tar.xz 文件名也是按自己的来。

3.加入读取权限

sudo chmod a+r /usr/local/cuda-11.7/include.cudnn.h
sudo chmod a+r /usr/local/cuda-11.7/lib64/libcudnn*
# 依次输入到终端

4.测试代码

cat /usr/local/cuda-11.7/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
# 输入到终端

输出即成功。

OK,这样你就完成了CUDA和cuDNN的安装啦,接下来去安装anaconda配置yolo环境吧~

标签: linux 运维 服务器

本文转载自: https://blog.csdn.net/2302_79753801/article/details/143031547
版权归原作者 蒋源昊 所有, 如有侵权,请联系我们删除。

“零基础也能Linux Ubuntu20.04版本安装CUDA和cuDNN,我教你!”的评论:

还没有评论