Ubuntu完善安装显卡驱动、CUDA、cuDNN

博主的Ubuntu版本为22.04.
  文章目录

• 一. Ubuntu安装显卡驱动
  
• 二. Ubuntu安装CUDA
  
• 三. Ubuntu安装cuDNN
  

一. Ubuntu安装显卡驱动

Ubuntu安装显卡驱动最方便的方法：打开*附加驱动*，直接安装选项的*利用NVIDIA...*即可，一般来说安装第一个就行了。

安装好重启后终端输入：

1. nvidia-smi

复制代码

有对应的输出则阐显着卡驱动安装乐成了。如果你的电脑是有核显和独显的，这个时候就能切换为核显正常进入Ubuntu系统了。

二. Ubuntu安装CUDA

上面输入nvidia-smi
后会有一个版本数字，我们要安装的CUDA版本≤显卡驱动版本。即我上面的显卡驱动版本为12.2，那么我可以安装小于便是12.2版本的CUDA，如果你比力后发现可以安装11.8的版本的CUDA，我强烈保举你按照这个版本。在不短的时间内，它是经典的不可撼动的存在，适配性很好。
打开官网：CUDA Toolkit Archive
找到CUDA Toolkit 11.8.0（如果你想安装别的版本的CUDA也是同理，选其他的即可），点击后根据你自己的电脑系统选择。博主的Ubuntu为22.04，选项供大家参考。

留意Installer Type选择runfile(local)，接下来在任一文件夹下的终端依次输入（如果你是选择安装的是其他版本的CUDA，安装命令在上图的下面官方会给出，把下面临应的命令替换掉即可）：

1. wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run

复制代码

1. sudo apt-get update

复制代码

1. sudo apt-get upgrade

复制代码

1. sudo apt install gcc

复制代码

1. sudo sh cuda_11.8.0_520.61.05_linux.run

复制代码

进入CUDA安装界面后按下面的选择操作，Driver和Kernel Objects不要勾选。移动到那个选项按Enter即可，之后Install.

接下来终端输入：

1. source ~/.bashrc

复制代码

在打开的文件中的最后写入：

1. export PATH=$PATH:/usr/local/cuda/bin

复制代码

1. export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

复制代码

1. export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64

复制代码

默认路径应该是这样，建议去上面的路径看看有没有。

保存退出即可。最后在终端输入：

1. nvcc -V

复制代码

有类似下图的输出则安装乐成。

三. Ubuntu安装cuDNN

打开官网：cuDNN Archive | NVIDIA Developer，留意下面的cuDNN有对应版本的CUDA，不要安装错了。比方Download cuDNN v8.9.7 (December 5th, 2023), for CUDA 12.x就是对应的是12.x的版本的CUDA。Download cuDNN v8.9.7 (December 5th, 2023), for CUDA 11.x就是对应的是11.x的版本的CUDA。博主安装的是CUDA11.8，这里我选择cuDNN v8.9.7进行安装。这里选择Local Installer for Linux x86_64 (Tar)，留意需要注册大概登录你的NVIDIA账号才能进行下载。

下载完成后参照官方安装步骤：Installing cuDNN on Linux — NVIDIA cuDNN v9.2.1 documentation，即下图。

我这边直接放出相关命令方便大家。先把下载下来的tar文件解压，之后在终端依次输入：

1. sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include

复制代码

1. sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64

复制代码

1. sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

复制代码

最后利用下面命令检测cuDNN是否乐成安装：

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

复制代码

有类似下面输出则乐成安装cuDNN，那三个数字就是你的版本号。

---

至此，Ubuntu完善安装显卡驱动、CUDA、cuDNN。

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。