当前位置:首页 > 系统教程 > 正文

Ubuntu 22.04深度学习环境搭建指南CUDA 11.8与cuDNN 8.6.0安装详细步骤(新手入门)

Ubuntu 22.04深度学习环境搭建指南CUDA 11.8与cuDNN 8.6.0安装详细步骤(新手入门)

欢迎来到深度学习环境搭建系列教程的第一篇。本文将手把手教你如何在Ubuntu 22.04系统上安装CUDA 11.8cuDNN 8.6.0,为后续的深度学习框架(如TensorFlow、PyTorch)打下坚实基础。无论你是刚入门的新手,还是希望梳理流程的老手,本文都将提供清晰、详细的步骤。

Ubuntu 22.04深度学习环境搭建指南CUDA 11.8与cuDNN 8.6.0安装详细步骤(新手入门) 深度学习环境搭建  CUDA 11.8安装 cuDNN 8.6.0配置 22.04教程 第1张

1. 系统要求与准备工作

在开始之前,请确保你的机器满足以下条件:

  • 已安装Ubuntu 22.04操作系统,并更新到最新。
  • 拥有NVIDIA显卡,且支持CUDA 11.8(计算能力3.5及以上)。
  • 具备基本的终端操作知识。
  • 稳定的网络连接以下载安装包。

2. 安装NVIDIA显卡驱动

CUDA依赖于NVIDIA驱动,因此需要先安装合适版本的驱动。推荐使用官方驱动PPA或直接安装驱动。步骤如下:

# 添加图形驱动PPAsudo add-apt-repository ppa:graphics-drivers/ppasudo apt update# 查看推荐的驱动版本ubuntu-drivers devices# 安装推荐驱动(例如nvidia-driver-525)sudo apt install nvidia-driver-525# 重启系统sudo reboot

重启后,运行 nvidia-smi 验证驱动是否安装成功。如果显示GPU信息,则驱动正常。

3. 安装CUDA 11.8

CUDA 11.8安装有多种方式,这里采用runfile安装,因为它更灵活且不依赖包管理器。首先,从NVIDIA官网下载CUDA 11.8 runfile:

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.runsudo sh cuda_11.8.0_520.61.05_linux.run

执行安装程序时,取消勾选Driver(因为我们已经安装了驱动),只安装CUDA Toolkit。按照提示完成安装。

4. 设置环境变量

安装完成后,需要将CUDA添加到PATH和LD_LIBRARY_PATH。编辑~/.bashrc:

echo "export PATH=/usr/local/cuda-11.8/bin:$PATH" >> ~/.bashrcecho "export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH" >> ~/.bashrcsource ~/.bashrc

5. 验证CUDA安装

运行以下命令检查CUDA是否安装成功:

nvcc -V

如果显示CUDA版本信息,则说明CUDA 11.8安装成功。

6. 安装cuDNN 8.6.0

cuDNN 8.6.0配置需要先下载cuDNN包。访问NVIDIA官网(需注册账号)下载cuDNN for CUDA 11.x的deb包或tar包。这里以tar包为例:

# 下载cudnn-linux-x86_64-8.6.0.163_cuda11-archive.tar.xztar -xvf cudnn-linux-x86_64-8.6.0.163_cuda11-archive.tar.xzsudo cp cudnn--archive/include/cudnn.h /usr/local/cuda-11.8/includesudo cp -P cudnn--archive/lib/libcudnn /usr/local/cuda-11.8/lib64sudo chmod a+r /usr/local/cuda-11.8/include/cudnn*.h /usr/local/cuda-11.8/lib64/libcudnn*

7. 验证cuDNN安装

通过查看cuDNN版本来验证:

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

如果显示版本号,则cuDNN安装成功。

8. 总结

至此,你已经在Ubuntu 22.04系统上成功搭建了包含CUDA 11.8和cuDNN 8.6.0的深度学习环境。接下来可以安装深度学习框架,开始你的模型训练之旅。本文详细介绍了每一步,即使是新手也能顺利完成深度学习环境搭建。后续我们将继续介绍更多环境的配置,敬请关注。

SEO关键词:深度学习环境搭建, CUDA 11.8安装, cuDNN 8.6.0配置, Ubuntu 22.04教程