admin管理员组

文章数量:1567746

文章目录

  • 一、Ubuntu 如何根据NVIDIA显卡型号确定对应的显卡驱动版本并安装
  • 二、根据安装的显卡驱动,安装对应的 CUDA 版本
    • 1. 查看显卡驱动版本
    • 2. 根据显卡驱动版本,安装对应的CUDA版本
      • (1) 确定 CUDA 版本
      • (2) 安装 CUDA
        • 1) 官网下载并安装 CUDA
        • 2) 直接二进制安装 CUDA
    • 3. 验证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/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/qq_43644413/article/details/124899366, 2022-05-24/2022-10-17.

本文标签: 显卡驱动UbuntuNVIDIACUDA