Ubuntu20.04 NVIDIA 显卡驱动与 CUDA、cudnn 安装指南

“ubuntu20.04显卡驱动,CUDA 安装和卸载,cudnn安装” 在Ubuntu 20.04操作系统中，针对NVIDIA显卡的配置涉及几个关键步骤，包括显卡驱动的安装、CUDA的安装与卸载以及cudnn的安装与检测。以下是对这些过程的详细说明： 一、显卡驱动安装 1-1 删除已安装的驱动 在升级或更换驱动前，需要确保没有遗留的旧版本驱动。通过执行`sudo apt purge nvidia*`命令，可以删除所有已安装的NVIDIA驱动。 1-2 禁用nouveau驱动 nouveau是Ubuntu默认的开源NVIDIA驱动。为了安装官方驱动，需要先禁用它。检查nouveau是否在运行，使用`lsmod | grep nouveau`命令。如果存在输出，表示nouveau正在运行。禁用方法是在`/etc/modprobe.d/blacklist.conf`文件末尾添加两行： ``` blacklist nouveau options nouveau modeset=0 ``` 然后执行`sudo update-initramfs -u`来应用改动，并重启系统。 1-3 安装NVIDIA驱动 从NVIDIA官方网站（https://www.nvidia.com/Download/index.aspx）下载对应版本的驱动，然后进行以下操作： 1. 设置驱动文件的执行权限，例如：`chmod +x NVIDIA-Linux-x86_64-510.54.run` 2. 在终端中运行该文件进行安装，过程中可能会遇到两个询问： - 如果出现关于预安装脚本失败的提示，选择"yes"继续。 - 当询问是否注册内核模块源到DKMS时，选择"NO"，避免自动构建新的模块。 二、CUDA安装和卸载 2-1 CUDA安装 安装CUDA需要首先从NVIDIA官网下载CUDA Toolkit。根据系统环境选择合适的版本，然后按照安装指南进行操作。通常步骤包括： 1. 解压下载的`.deb`包，如`cuda_11.4.2_460.32.03_linux.deb`。 2. 使用`sudo dpkg -i <filename>.deb`安装包。 3. 添加环境变量到`.bashrc`或`.bash_profile`，例如： ``` export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH ``` 4. 源文件更新环境变量：`source ~/.bashrc` 5. 验证CUDA安装：`nvcc --version` 2-2 CUDA卸载 要卸载CUDA，可使用以下命令： 1. 移除CUDA软件包：`sudo apt-get remove --purge cuda-<version>`（替换<version>为实际版本号） 2. 清理相关依赖：`sudo apt-get autoremove` 3. 更新环境变量，移除添加的CUDA路径。 三、cudnn安装与检测 3-1 cudnn安装 cudnn是CUDA的深度学习库。下载cudnn压缩包后，解压并将头文件和库文件复制到CUDA安装目录： 1. 解压cudnn文件，如`cuda<version>-<release>.tar.gz`。 2. 将解压后的`include`和`lib64`目录下的内容分别复制到`/usr/local/cuda/include`和`/usr/local/cuda/lib64`。 3. 更新库文件索引：`sudo ldconfig` 3-2 cudnn检测 验证cudnn是否安装成功，可以编译一个简单的示例程序，比如从cudnn样例代码中选取，或者使用以下命令检查cudnn版本： ```bash cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 ``` 输出应显示cudnn的版本信息。 以上就是Ubuntu 20.04系统中NVIDIA显卡驱动、CUDA及cudnn的安装、卸载与检测的完整流程。确保每个步骤都按顺序正确执行，以保证系统稳定性和性能优化。在操作过程中，遇到问题可以查阅官方文档或社区支持来解决问题。