前言
用于深度学习的环境配置过程:新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环境吧~
版权归原作者 蒋源昊 所有, 如有侵权,请联系我们删除。