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

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

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

本教程将详细介绍在Ubuntu 20.04系统上安装NVIDIA驱动、CUDA和cuDNN的完整步骤,专为小白设计,避免常见坑点,并附图文详解。通过本指南,您可以顺利配置深度学习环境。

一、准备工作

在开始安装之前,请确保您的系统是Ubuntu 20.04,并且拥有NVIDIA显卡。首先,更新系统包列表以获取最新软件信息:

    sudo apt updatesudo apt upgrade  

接下来,禁用默认的Nouveau驱动,这是避免驱动冲突的关键步骤。编辑黑名单文件:

    sudo nano /etc/modprobe.d/blacklist-nouveau.conf  

在文件中添加以下内容:

    blacklist nouveauoptions nouveau modeset=0  

保存文件后,更新initramfs并重启系统:

    sudo update-initramfs -usudo reboot  
全网最详细Ubuntu20.04安装NVIDIA驱动+CUDA+cuDNN教程(避坑指南与图文详解) Ubuntu 20.04  NVIDIA驱动安装 CUDA安装 cuDNN安装 第1张

二、安装NVIDIA驱动

完成准备工作后,开始NVIDIA驱动安装。推荐使用PPA仓库安装,以确保驱动兼容性。首先,添加PPA仓库:

    sudo add-apt-repository ppa:graphics-drivers/ppasudo apt update  

然后,查看可用的驱动版本:

    ubuntu-drivers devices  

选择推荐版本进行安装,例如nvidia-driver-450:

    sudo apt install nvidia-driver-450  

安装完成后,重启系统以应用驱动:

    sudo reboot  

重启后,验证驱动是否安装成功:

    nvidia-smi  

如果输出GPU信息,则NVIDIA驱动安装成功。

三、安装CUDA

接下来进行CUDA安装。CUDA是NVIDIA的并行计算平台,需从官网下载对应版本。对于Ubuntu 20.04,建议使用runfile安装。下载后,运行安装文件(以CUDA 11.0为例):

    sudo sh cuda_11.0.2_450.51.05_linux.run  

在安装过程中,务必选择不安装驱动,以免覆盖现有驱动。安装完成后,配置环境变量:

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

验证CUDA安装

    nvcc -V  

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

四、安装cuDNN

最后进行cuDNN安装。cuDNN是深度神经网络库,需从NVIDIA开发者网站下载与CUDA兼容的版本。下载后,解压并复制文件:

    tar -xzvf cudnn-11.0-linux-x64-v8.0.5.39.tgzsudo cp cuda/include/cudnn.h /usr/local/cuda/includesudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*  

验证cuDNN安装

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

如果输出版本信息,则cuDNN安装成功。

五、常见问题避坑

1. 黑屏或循环登录:可能是驱动冲突,请确保已彻底禁用Nouveau驱动。

2. CUDA安装失败:检查驱动兼容性,并确保在安装时选择不安装驱动。

3. 版本不匹配:CUDA和cuDNN版本需兼容,请参考NVIDIA官方文档。

通过本教程,您应已成功在Ubuntu 20.04上完成NVIDIA驱动、CUDA和cuDNN的安装。如有问题,欢迎在评论区交流。