深度学习环境配置——ubuntu安装CUDA与CUDNN

目录

• Ⅰ. 前言
• Ⅱ. 查看显卡驱动兼容性
• Ⅲ. 安装CUDA 12.1
  • 3.1 CUDA下载
  • 3.2 CUDA安装
  • 3.3 添加环境变量
• Ⅳ. ubuntu安装CUDNN
  • 4.1 下载
  • 4.2 安装
  • 4.3 版本查看

操作系统： ubuntu 20.04
更新时间： 2024年2月21日
显卡驱动版本： 535.154
安装说明： 电脑必须要有 NVIDIA 独立显卡，且已经安装显卡驱动；全局安装，非Conda虚拟环境

关注微信公众号：XRobotSpace
关注微信公众号：依法编程

Ⅰ. 前言

CUDA的安装需要两个包：CUDA和CUDNN（CUDA的一个补丁，用于优化深度学习），先安装CUDA，后安装CUDNN。

CUDA工具包发行说明：
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

CUDA版本汇总：
https://developer.nvidia.com/cuda-toolkit-archive

CUDA安装官方教程：
https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html

CUDNN安装官方教程：
https://docs.nvidia.com/deeplearning/cudnn/installation/windows.html

Ⅱ. 查看显卡驱动兼容性

• GPU信息、驱动版本可以通过cmd输入nvidia-smi指令来查看。
• 显卡驱动版本向下兼容，其决定了可安装的CUDA Toolkit的最高版本，高版本的驱动支持低版本的CUDA。

  https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
  

• 比如我的显卡驱动版本是535.154，可安装的CUDA最高版本为12.2。（可能是推荐版本，未测试）
  

Ⅲ. 安装CUDA 12.1

3.1 CUDA下载

wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run

3.2 CUDA安装

sudo sh cuda_12.1.0_530.30.02_linux.run

• 输入accept。

• 去掉Driver勾选，因为已经安装了显卡驱动，不需要覆盖安装。

• 安装完成

3.3 添加环境变量

.bashrc文件中添加内容。（一台电脑可以安装多个CUDA版本，修改.bashrc文件实现不同版本切换。）

export PATH=/usr/local/cuda-12.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

Ⅳ. ubuntu安装CUDNN

4.1 下载

# 版本合集
https://developer.nvidia.com/cudnn-archive

# cuDNN 9.0.0
https://developer.nvidia.com/cudnn-downloads?target_os=Linux&target_arch=x86_64&Distribution=Agnostic&cuda_version=12

• 下载压缩包

4.2 安装

• 拷贝文件到制定目录+授予权限命令

sudo cp cudnn/include/cudnn* /usr/local/cuda-12.1/include
sudo cp cudnn/lib/libcudnn* /usr/local/cuda-12.1/lib64
sudo chmod a+r /usr/local/cuda-12.1/include/cudnn* 
sudo chmod a+r /usr/local/cuda-12.1/lib64/libcudnn*

4.3 版本查看

• 查看CUDNN版本

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2