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

深度学习环境搭建运行(一):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.8和CUDNN 8.6.0的步骤,帮助小白快速搭建深度学习环境。我们会从系统检查开始,逐步完成驱动安装、CUDA配置和CUDNN部署,确保你能顺利运行深度学习项目。

1. 系统要求检查

首先,确保你的系统是Ubuntu 22.04,这是深度学习环境搭建的基础。打开终端(Ctrl+Alt+T),运行以下命令检查系统版本:

    lsb_release -a  

同时,检查NVIDIA显卡信息,确保支持CUDA:

    nvidia-smi  

如果未安装驱动,请继续下一步。否则,可直接跳过驱动安装部分。

2. 安装NVIDIA驱动

在Ubuntu 22.04上,推荐使用APT安装NVIDIA驱动。运行以下命令:

    sudo apt updatesudo apt install nvidia-driver-535  

安装完成后,重启系统使驱动生效:

    sudo reboot  

3. 安装CUDA 11.8

CUDA 11.8安装是深度学习环境搭建的核心步骤。从NVIDIA官网下载安装包,或使用wget命令:

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

运行安装程序,按照提示操作。注意在安装选项中取消勾选NVIDIA驱动(因已安装):

    sudo sh cuda_11.8.0_520.61.05_linux.run  

安装完成后,配置环境变量。编辑~/.bashrc文件:

    nano ~/.bashrc  

在文件末尾添加以下行,确保CUDA路径正确:

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

保存并退出,然后应用更改:

    source ~/.bashrc  

验证CUDA 11.8安装是否成功:

    nvcc --version  

如果输出显示CUDA 11.8,则安装完成。这里插入一张示意图展示验证过程:

深度学习环境搭建运行(一):Ubuntu 22.04系统安装CUDA 11.8和CUDNN 8.6.0详细步骤 深度学习环境搭建 Ubuntu 22.04 CUDA 11.8安装 CUDNN 8.6.0配置 第1张

4. 安装CUDNN 8.6.0

CUDNN 8.6.0配置是深度学习环境搭建的最后关键步骤。从NVIDIA官网下载对应CUDA 11.8的CUDNN文件(需注册)。解压并复制文件:

    tar -xzvf cudnn-11.8-linux-x64-v8.6.0.163.tgzsudo cp cuda/include/cudnn*.h /usr/local/cuda-11.8/includesudo cp cuda/lib64/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*  

验证CUDNN安装,检查版本信息:

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

应输出版本8.6.0,表示CUDNN配置成功。

5. 验证深度学习环境

完成CUDA 11.8安装和CUDNN 8.6.0配置后,可测试深度学习环境。例如,安装Python和TensorFlow,运行GPU加速代码。这确保了Ubuntu 22.04系统上的深度学习环境搭建完整。

总结:本教程详细讲解了在Ubuntu 22.04系统上安装CUDA 11.8和CUDNN 8.6.0的步骤,覆盖了从驱动安装到环境验证的全过程。通过遵循本指南,即使新手也能顺利完成深度学习环境搭建,为后续项目运行奠定基础。