Ubuntu 安装NVIDIA显卡驱动、安装对应的CUDA

一、Ubuntu 如何根据NVIDIA显卡型号确定对应的显卡驱动版本并安装

Ubuntu 如何根据NVIDIA显卡型号确定对应的显卡驱动版本并安装

二、根据安装的显卡驱动，安装对应的 CUDA 版本

1. 查看显卡驱动版本

通过终端 nvidia-smi查看

nvidia-smi  #查看GPU信息, 安装完显卡驱动后需要先重启一次电脑

可以发现，我这里的NVIDIA显卡驱动版本是 470.141.03，这里可以看到本机最高支持 cuda 11.4 。.

2. 根据显卡驱动版本，安装对应的CUDA版本

(1) 确定 CUDA 版本

所以需要根据已有的显卡驱动版本，去NVIDIA官网查看该驱动版本所能支持的CUDA版本：
如下表：

通过上表可以发现，由于自己 Ubuntu 下的NVIDIA显卡驱动版本是 470.141.03，所以自己电脑显卡最高支持的是 CUDA 11.4 Update 4 .

(2) 安装 CUDA

1) 官网下载并安装 CUDA

从官网下载CUDA：
下载最新版 CUDA

下载 历史版本的CUDA

然后选择自己对应的系统版本的CUDA，并下载：

按照指令一个一个执行：

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run
sudo sh cuda_11.4.0_470.42.01_linux.run

选择 Continue ，然后输入 accept：

按回车取消 Driver 的安装（因为一般已经装好驱动了），然后回车选择 Install：

上面一步，选择 Install 后,终端就会跳出选择界面，回到正常终端模式，并且没有用任何提示。这是正常现象，等待安装完成即可。安装完成后，会出现如下类似界面：

添加环境变量，sudo gedit ~/.bashrc ，在末尾添加：

export PATH=$PATH:/usr/local/cuda-11.4/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.4/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.4/lib64

2) 直接二进制安装 CUDA

直接二进制安装的版本版本号会比较低。

sudo apt-get install nvidia-cuda-toolkit

安装包比较大，等待安装完成。

3. 验证CUDA是否安装成功

nvcc -V   # 或者 nvcc --version

如果输出以下内容，说明 CUDA 安装成功。

参考链接

[1] FAST-ROBOT. ubuntu18.04安装cuda和cudnn [EB/OL]. https://blog.csdn.net/qq_43644413/article/details/124899366, 2022-05-24/2022-10-17.