Ubuntu22.04深度学习环境安装【cuda+cudnn】
为了复现一篇深度学习论文，特意安装了Linux系统。前一天已经安装Linux显卡驱动，如今必要安装cuda、cudnn等。论文代码
论文PDF
确定包版本：
根据论文提供的代码。在requirements.txt中发现cuda版本为11.7,cudnn为8.5.0，python没有指定，默认使用3.8版本，pytorch2.0.1。接下来就分别安装这些包：
https://i-blog.csdnimg.cn/direct/96fa97141f4849bab84af34191b80c6d.png
一、安装cuda
[*] 下载cuda11.7安装程序
cuda汗青版本官网：cuda下载
https://i-blog.csdnimg.cn/direct/541f319c204e45f48a72d34f73481c9d.png
输入以下命令下载cuda11.7：
wget https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda_11.7.1_515.65.01_linux.run
[*] 运行cuda11.7安装程序：
sudo sh cuda_11.7.1_515.65.01_linux.run
[*] 解决报错
报错信息如下：
https://i-blog.csdnimg.cn/direct/19097de919b248828c9273498f9293de.png
检察报错日志文件：
gedit /var/log/cuda-installer.log
https://i-blog.csdnimg.cn/direct/64d8f568154a4f07a579f17d1fb00946.png
发现是大概是gcc版本过高，可以看我之前的博客：切换gcc版本
检察当前gcc版本和优先级：
sudo update-alternatives --config gcc
https://i-blog.csdnimg.cn/direct/5d7a2522cad644d184bb9aeb79792d2e.png
修改gcc-11优先级：
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-11 102
https://i-blog.csdnimg.cn/direct/664cd51c048948c383047ac9e717da02.png
[*] 继承运行cuda11.7安装程序：
sudo sh cuda_11.7.1_515.65.01_linux.run
选择accept
https://i-blog.csdnimg.cn/direct/76ee3d637b4f4a1981aa42b0b9602ebb.png
取消驱动安装（空格取消）,由于已经安装好了驱动了。
https://i-blog.csdnimg.cn/direct/bec39275b27c4688be806e8fb41ac2e7.png
以下体现安装完成：
https://i-blog.csdnimg.cn/direct/4ab0488e98024309a11fb95a9a6e079d.png
此时输入nvcc --version
不能检察版本，由于没有设置环境变量
[*] 设置环境变量
gedit ~/.bashrc
打开的文件：
https://i-blog.csdnimg.cn/direct/1466c548571147d988730318f842965a.png
在文件末端添加并保存以下内容：（注意cuda版本要同等）
export PATH=/usr/local/cuda-11.7/bin:${PATH}
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64:${LD_LIBRARY_PATH}
终端输入：
source ~/.bashrc
[*] 验证cuda环境变量
nvcc --version
https://i-blog.csdnimg.cn/direct/01d63111a8ea4be5a66465f4e8ce6fc2.png
二、安装cudnn
[*] 下载cudnn文件（必要登岸Nvidia账号）
官网：cudnn下载
https://i-blog.csdnimg.cn/direct/810366fe0ae94be7a3f70c44d45a9889.png
由于要复现的论文使用cudnn8.5.0版本，以是就选择这个版本了。
https://i-blog.csdnimg.cn/direct/9042a109205540e989609f8c318386e3.png
[*] 解压文件
tar -xvf cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz
[*] 将cudnn文件复制到cudn文件中
cd cudnn-linux-x86_64-8.5.0.96_cuda11-archive/
sudo cp include/cudnn.h /usr/local/cuda-11.7/include
sudo cp lib/libcudnn* /usr/local/cuda-11.7/lib64
sudo chmod a+r /usr/local/cuda-11.7/include/cudnn.h /usr/local/cuda-11.7/lib64/libcudnn*
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。
页:
[1]