在使用Ubuntu操作系统时，安装显卡驱动、CUDA、cuDNN是非常重要的步骤，特别是对于进行深度学习、机器学习和图形渲染等任务的用户来说。下面将为大家一步一步详细介绍如何在Ubuntu上进行显卡驱动、CUDA和cuDNN的安装。
安装显卡驱动
第一步，打开终端，运行以下命令更新系统软件包：
sudo apt-get update
sudo apt-get upgrade
第二步，运行以下命令以检测可用的显卡驱动版本：
ubuntu-drivers devices
第三步，在获取到推荐的显卡驱动版本后，运行以下命令进行安装：
sudo apt-get install nvidia-driver-{VERSION}
其中，{VERSION}是获取到的显卡驱动版本号。
第四步，安装完成后，重新启动计算机。可以运行以下命令检查驱动是否已成功安装：
nvidia-smi
如果能够正常显示显卡信息，则说明驱动已成功安装。
CUDA安装
第一步，前往NVIDIA官方网站（https://developer.nvidia.com/cuda-downloads）下载所需版本的CUDA。选择适用于Ubuntu操作系统的版本，下载完成后进行安装。
sudo dpkg -i cuda-repo-ubuntu{VERSION}_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu{VERSION}/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
其中，{VERSION}是Ubuntu操作系统的版本。
第二步，完成安装后，需要配置环境变量。运行以下命令打开.bashrc文件：
sudo nano ~/.bashrc
在文件末尾添加以下内容：
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
保存并退出文件。
第三步，重新加载.bashrc文件使环境变量生效：
source ~/.bashrc
第四步，运行以下命令验证CUDA安装是否成功：
nvcc --version
如果能够正常显示CUDA的版本信息，则说明安装已成功。
cuDNN安装
第一步，前往NVIDIA官方网站（https://developer.nvidia.com/cudnn）下载所需版本的cuDNN。选择适用于自己安装的CUDA版本的cuDNN，下载完成后进行安装。
第二步，解压下载的cuDNN压缩包并进入文件夹：
tar -xzvf cudnn-10.2-linux-x64-v8.0.2.39.tgz
cd cuda
其中，cudnn-10.2-linux-x64-v8.0.2.39.tgz是下载的cuDNN压缩包文件名，具体根据自己下载的版本来命名。
第三步，将cuDNN的文件复制到CUDA的安装目录：
sudo cp include/cudnn.h /usr/local/cuda/include
sudo cp lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
第四步，运行以下命令以刷新系统库缓存：
sudo ldconfig
第五步，cuDNN安装完成。可以运行以下命令验证安装是否成功：
cat /usr/local/cuda/include/cudnn.h | grep -E "CUDNN_MAJOR|CUDNN_MINOR|CUDNN_PATCHLEVEL"
如果能够正常显示cuDNN的版本信息，则说明安装已成功。
通过以上步骤，我们可以在Ubuntu操作系统上成功安装显卡驱动、CUDA和cuDNN，为后续深度学习任务的运行和开发提供良好的环境支持。希望这篇博客对你有所帮助！
本文来自极简博客，作者：编程狂想曲，转载请注明原文链接：Ubuntu显卡驱动安装（更新）- CUDA安装- cuDNN安装
