运维系列&AI系列&Lenovo-G双系统系列(亲测有效):Ubuntu 安装NVIDIA显卡驱动、安装对应的CUDA




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.







wongHome

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

posted @ 2024-08-15 23:43  坦笑&&life  阅读(14)  评论(0编辑  收藏  举报  来源