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

RTX 5060 Max-Q 显卡 Ubuntu 20.04 驱动安装教程 (CUDA 11.4 + cuDNN 8.9.5 深度学习环境配置指南)

RTX 5060 Max-Q 显卡 Ubuntu 20.04 驱动安装教程 (CUDA 11.4 + cuDNN 8.9.5 深度学习环境配置指南)

SEO关键词:RTX 5060驱动安装、Ubuntu 20.04 CUDA 11.4、cuDNN 8.9.5 配置、深度学习环境搭建

对于刚接触 Linux 的深度学习开发者来说,在 Ubuntu 20.04 系统下配置显卡驱动、CUDA 和 cuDNN 是一道必须跨过的坎。本文将针对 RTX 5060 Max-Q 显卡,详细介绍如何正确安装显卡驱动、CUDA 11.4 以及 cuDNN 8.9.5,确保您的开发环境稳定高效。

RTX 5060 Max-Q 显卡 Ubuntu 20.04 驱动安装教程 (CUDA 11.4 + cuDNN 8.9.5 深度学习环境配置指南) 5060驱动安装  CUDA 配置 深度学习环境搭建 第1张

一、 环境准备与旧驱动清理

在开始之前,我们需要确保系统环境干净,避免版本冲突。执行以下命令更新系统并卸载可能存在的旧驱动:

sudo apt-get updatesudo apt-get upgradesudo apt-get remove --purge nvidia*

二、 安装 NVIDIA 显卡驱动

对于 RTX 5060驱动安装,建议使用 PPA 源安装,这种方式最为简便且稳定:

  1. 添加驱动源:sudo add-apt-repository ppa:graphics-drivers/ppa
  2. 更新源:sudo apt update
  3. 查看推荐驱动:ubuntu-drivers devices
  4. 安装驱动(建议选择 535 或更高版本以适配新卡):sudo apt install nvidia-driver-535

安装完成后,请务必重启电脑:sudo reboot。重启后输入 nvidia-smi 检查是否显示显卡信息。

三、 安装 CUDA 11.4

Ubuntu 20.04 CUDA 11.4 的安装建议前往 NVIDIA 官网下载 runfile 文件:

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.runsudo sh cuda_11.4.0_470.42.01_linux.run

安装过程中,注意在勾选界面取消选择 Driver(因为我们已经手动安装了驱动),只安装 Toolkit 和 Samples。

四、 配置环境变量

打开终端,输入 gedit ~/.bashrc,在末尾添加:

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

保存并退出,执行 source ~/.bashrc 使其生效。

五、 cuDNN 8.9.5 配置

cuDNN 8.9.5 配置 需要下载对应的 Tar 压缩包,解压后将文件拷贝到 CUDA 安装目录:

# 解压后执行sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda-11.4/includesudo cp cudnn-*-archive/lib/libcudnn* /usr/local/cuda-11.4/lib64sudo chmod a+r /usr/local/cuda-11.4/include/cudnn*.h /usr/local/cuda-11.4/lib64/libcudnn*

六、 验证安装

至此,您的深度学习环境搭建已基本完成。验证 CUDA 状态:

nvcc -V

如果显示 CUDA 11.4 的版本信息,恭喜您,环境配置成功!