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

全网最详细Ubuntu20.04安装NVIDIA驱动+CUDA+cuDNN教程 (避坑指南|图文详解)

全网最详细Ubuntu20.04安装NVIDIA驱动+CUDA+cuDNN教程 (避坑指南|图文详解)

⚠️ 小白必看: 本文旨在手把手教你如何在 Ubuntu20.04 上完整配置深度学习环境,包括 NVIDIA驱动CUDAcuDNN。全程避坑,图文并茂,即使零基础也能成功!


  全网最详细Ubuntu20.04安装NVIDIA驱动+CUDA+cuDNN教程 (避坑指南|图文详解)
 NVIDIA驱动安装 CUDA安装教程 cuDNN配置 Ubuntu20.04深度学习环境配置 第1张

🔍 第一步:检查硬件与系统准备

打开终端 (Ctrl+Alt+T),输入以下命令查看显卡型号:

lspci | grep -i nvidia

若看到类似 NVIDIA Corporation GA102 [GeForce RTX 3080] 的字段,说明NVIDIA显卡已识别。接着确认系统是否支持:

uname -m && cat /etc/os-release

确保显示 x86_64Ubuntu 20.04

🚫 第二步:禁用系统默认驱动 nouveau (必做!)

Ubuntu默认集成的开源驱动 nouveau 会与NVIDIA闭源驱动冲突,必须禁用。执行:

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

✅ 重启后验证:lsmod | grep nouveau 无输出则禁用成功。

⚙️ 第三步:安装NVIDIA驱动 (三种方法,推荐方法二)

方法一: 自动推荐安装(最稳)

ubuntu-drivers devices          # 查看可用驱动版本sudo apt install nvidia-driver-470   # 选择 recommended 版本

方法二: 从NVIDIA官网下载.run文件(可精确选择版本,适合有特殊需求)

wget https://us.download.nvidia.com/XFree86/Linux-x86_64/550.54.14/NVIDIA-Linux-x86_64-550.54.14.runchmod +x NVIDIA-Linux-x86_64-*.runsudo ./NVIDIA-Linux-x86_64-*.run

安装后重启,输入 nvidia-smi 查看驱动信息,若显示GPU列表则 NVIDIA驱动安装 成功。

📦 第四步:安装CUDA (与驱动版本匹配)

访问 CUDA Toolkit Archive 选择与驱动兼容的版本(例如驱动550.x推荐CUDA 12.x)。以CUDA 12.2为例:

wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.runsudo sh cuda_12.2.0_535.54.03_linux.run

安装时取消勾选 Driver (因为已安装驱动),仅安装CUDA Toolkit。配置环境变量:

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

验证:nvcc -V 显示版本信息即 CUDA安装教程 完成。

🧩 第五步:配置cuDNN (深度学习加速库)

前往 NVIDIA cuDNN 注册下载对应CUDA 12.2的cuDNN版本(需登录)。下载 Local Installer for Ubuntu20.04 x86_64 (Deb),然后:

sudo dpkg -i cudnn-local-repo-ubuntu2004-8.9.5.30_1.0-1_amd64.debsudo cp /var/cudnn-local-repo-/cudnn-local--keyring.gpg /usr/share/keyrings/sudo apt updatesudo apt install libcudnn8 libcudnn8-dev libcudnn8-samples

验证安装:

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

✅ 看到 Test passed! 表示 cuDNN配置 成功!

🎯 总结与避坑提醒

至此,Ubuntu20.04深度学习环境配置 全部完成!核心避坑点:1) 务必禁用nouveau;2) 驱动与CUDA版本对应;3) 安装CUDA时取消驱动勾选;4) cuDNN需注册下载。如果遇到问题,欢迎留言交流。

⭐️ 本文关键词:NVIDIA驱动安装CUDA安装教程cuDNN配置Ubuntu20.04深度学习环境配置