Ubuntu 安装CUDA

本文主要介绍Ubuntu系统下如何安装Nvidia显卡驱动以及安装对应的CUDA (面向深度学习用户)

一、在安装CUDA之前需要安装显卡驱动，参考下方博客【若已安装请忽略】

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

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

1. 查看显卡驱动版本

通过终端 nvidia-smi查看

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

可以发现，我这里的NVIDIA显卡驱动版本是 470.141.03，这里可以看到本机最高支持 cuda 11.4 ，也就是说11.4之下的CUDA版本该显卡都可以支持，11.4之上的则不支持。

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

(1) 确定 CUDA 版本

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

通过上表可以发现，由于自己的NVIDIA显卡驱动版本是 470.141.03，所以自己电脑显卡支持的版本CUDA 11.4及之下。

(2) 安装 CUDA（介绍两种安装：安装最新版与安装历史版）

下载最新版 CUDA

下载历史版本的CUDA（以CUDA11.4为例）

然后选择自己对应的系统版本的CUDA，并下载（在这里我们推荐用runfile(local)的方式进行下载）：
请注意，有的cuda版本只支持特定版本的linux发行版系统，请进行匹对。

按照指令依次执行：

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

选择 Continue ，然后输入 accept：

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

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

添加环境变量，

sudo vim ~/.bashrc

，在其中添加并保存：

exportPATH=/usr/local/cuda-11.4/bin:$PATH

注意，其中的cuda版本选为你要安装的版本，上面命令中的是cuda-11.4，请根据实际情况进行更改。
接着：

source ~/.bashrc

至此我们安装cuda成功，可以通过以下命令查看cuda信息：
查看CUDA版本信息：

nvcc --version