当前位置： 首页>大数据>正文

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

大数据2024-10-21 21:18:21

深度学习环境配置——ubuntu安装CUDA与CUDNN,在这里插入图片描述,第1张

目录

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

Ⅰ. 前言

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。（可能是推荐版本，未测试）
    深度学习环境配置——ubuntu安装CUDA与CUDNN,第2张

深度学习环境配置——ubuntu安装CUDA与CUDNN,第3张

Ⅲ. 安装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

深度学习环境配置——ubuntu安装CUDA与CUDNN,第4张

3.2 CUDA安装

sudo sh cuda_12.1.0_530.30.02_linux.run
  • 输入accept。

深度学习环境配置——ubuntu安装CUDA与CUDNN,第5张

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

深度学习环境配置——ubuntu安装CUDA与CUDNN,第6张

  • 安装完成

深度学习环境配置——ubuntu安装CUDA与CUDNN,第7张

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
  • 下载压缩包

深度学习环境配置——ubuntu安装CUDA与CUDNN,第8张

4.2 安装

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

深度学习环境配置——ubuntu安装CUDA与CUDNN,第9张

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

https://www.xamrdz.com/bigdata/7ca1996001.html

相关文章：