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

RTX5060 Ubuntu20.04 驱动完美安装与CUDA11.4+cudnn8.9.5配置指南(小白避坑详细版)

RTX5060 Ubuntu20.04 驱动完美安装与CUDA11.4+cudnn8.9.5配置指南(小白避坑详细版)

RTX5060 Ubuntu20.04 驱动完美安装与CUDA11.4+cudnn8.9.5配置指南(小白避坑详细版) RTX5060驱动安装 CUDA11.4 cudnn8.9.5配置 显卡驱动正确流程 第1张

1. 准备工作:系统环境检查

首先确保你的Ubuntu20.04系统已更新,且拥有NVIDIA RTX5060显卡。打开终端,执行以下命令查看当前显卡信息:

lspci | grep -i nvidia

如果能看到NVIDIA相关设备,则继续。另外确认系统已安装gcc、make等编译工具:

sudo apt updatesudo apt install build-essential

2. 卸载旧版NVIDIA驱动(避免冲突)

如果之前安装过NVIDIA驱动,建议彻底卸载:

sudo apt purge nvidia*  sudo apt autoremove

3. 禁用nouveau开源驱动

nouveau会与官方驱动冲突,必须禁用:

sudo bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nvidia-nouveau.conf"sudo bash -c "echo options nouveau modeset=0 >> /etc/modprobe.d/blacklist-nvidia-nouveau.conf"

更新内核并重启:

sudo update-initramfs -usudo reboot

4. 下载RTX5060对应驱动

访问NVIDIA官网下载Linux驱动(.run文件),或使用命令行下载最新版。RTX5060需驱动版本≥535,这里以550为例:

wget https://us.download.nvidia.com/XFree86/Linux-x86_64/550.90.07/NVIDIA-Linux-x86_64-550.90.07.run

5. 安装驱动

停止图形界面(按Ctrl+Alt+F2进入纯终端,登录后执行):

sudo service lightdm stop   # 如果使用lightdmsudo init 3

给下载的驱动添加执行权限并安装:

chmod +x NVIDIA-Linux-x86_64-*.runsudo ./NVIDIA-Linux-x86_64-*.run

安装过程中如果提示32位兼容库,选择Yes;如果提示已有驱动,选Continue installation。安装完成后重启:

sudo reboot

重启后验证驱动:

nvidia-smi

如果显示GPU信息,则驱动安装成功。

6. 安装CUDA 11.4

访问NVIDIA CUDA Toolkit Archive,选择CUDA Toolkit 11.4.4,对应Ubuntu20.04的runfile或deb方式。推荐使用runfile:

wget https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux.runsudo sh cuda_11.4.4_470.82.01_linux.run

安装时取消勾选Driver(因为驱动已装),仅安装CUDA Toolkit。记录安装路径(默认/usr/local/cuda-11.4)。

7. 配置CUDA环境变量

编辑~/.bashrc:

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

验证CUDA版本:

nvcc -V

8. 安装cuDNN 8.9.5

从NVIDIA官网下载cuDNN for CUDA 11.x(需要注册账号)。下载三个deb文件:

  • libcudnn8_8.9.5.xx-1+cuda11.8_amd64.deb
  • libcudnn8-dev_8.9.5.xx-1+cuda11.8_amd64.deb
  • libcudnn8-samples_8.9.5.xx-1+cuda11.8_amd64.deb

注意选择与CUDA11兼容的版本,cuDNN 8.9.5 for CUDA 11.x均可。按顺序安装:

sudo dpkg -i libcudnn8_8.9.5.*_amd64.debsudo dpkg -i libcudnn8-dev_8.9.5.*_amd64.debsudo dpkg -i libcudnn8-samples_8.9.5.*_amd64.deb

9. 验证cuDNN安装

复制示例并编译测试:

cp -r /usr/src/cudnn_samples_v8/ $HOMEcd $HOME/cudnn_samples_v8/mnistCUDNNmake clean && make./mnistCUDNN

如果看到Test passed!,则cuDNN安装成功。

10. 常见踩坑点

• 驱动安装后黑屏:可能是nouveau未禁用,重新进入恢复模式禁用。 • CUDA版本与驱动不兼容:确保驱动版本≥470,最好使用550+。 • cuDNN库找不到:检查LD_LIBRARY_PATH是否包含cuda/lib64。 • 使用gcc版本过高:CUDA 11.4最高支持gcc-10,可降级gcc:sudo apt install gcc-9 g++-9。

至此,RTX5060显卡驱动、CUDA11.4和cuDNN8.9.5已全部正确安装!