Ubuntu22.04深度学习环境安装【cuda+cudnn】

为了复现一篇深度学习论文，特意安装了Linux系统。前一天已经安装Linux显卡驱动，如今必要安装cuda、cudnn等。
论文代码
论文PDF
确定包版本：
根据论文提供的代码。在requirements.txt中发现cuda版本为11.7,cudnn为8.5.0，python没有指定，默认使用3.8版本，pytorch2.0.1。接下来就分别安装这些包：

一、安装cuda

• 下载cuda11.7安装程序
  cuda汗青版本官网：cuda下载
  
  
  
  输入以下命令下载cuda11.7：
  
  1. wget https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda_11.7.1_515.65.01_linux.run

  复制代码
• 运行cuda11.7安装程序：
  
  1. sudo sh cuda_11.7.1_515.65.01_linux.run

  复制代码
• 解决报错
  报错信息如下：
  
  
  
  检察报错日志文件：
  
  1. gedit /var/log/cuda-installer.log

  复制代码
  
  
  发现是大概是gcc版本过高，可以看我之前的博客：切换gcc版本
  检察当前gcc版本和优先级：
  
  1. sudo update-alternatives --config gcc

  复制代码
  
  
  修改gcc-11优先级：
  
  1. sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-11 102

  复制代码
  
  
  
• 继承运行cuda11.7安装程序：
  
  1. sudo sh cuda_11.7.1_515.65.01_linux.run

  复制代码
  选择accept
  
  
  
  取消驱动安装（空格取消）,由于已经安装好了驱动了。
  
  
  
  以下体现安装完成：
  
  
  
  此时输入nvcc --version
  不能检察版本，由于没有设置环境变量
  
• 设置环境变量
  
  1. gedit ~/.bashrc

  复制代码
  打开的文件：
  
  
  
  在文件末端添加并保存以下内容：（注意cuda版本要同等）
  
  1. export PATH=/usr/local/cuda-11.7/bin:${PATH}
  2. export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64:${LD_LIBRARY_PATH}

  复制代码
  终端输入：
  
  1. source ~/.bashrc

  复制代码
• 验证cuda环境变量
  
  1. nvcc --version

  复制代码
  
  
  

二、安装cudnn

• 下载cudnn文件（必要登岸Nvidia账号）
  官网：cudnn下载
  
  
  
  由于要复现的论文使用cudnn8.5.0版本，以是就选择这个版本了。
  
  
  
  
• 解压文件
  
  1. tar -xvf cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz

  复制代码
• 将cudnn文件复制到cudn文件中
  
  1. cd cudnn-linux-x86_64-8.5.0.96_cuda11-archive/
  2. sudo cp include/cudnn.h /usr/local/cuda-11.7/include
  3. sudo cp lib/libcudnn* /usr/local/cuda-11.7/lib64
  4. sudo chmod a+r /usr/local/cuda-11.7/include/cudnn.h /usr/local/cuda-11.7/lib64/libcudnn*

  复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。