在Ubuntu16/18上安装CUDA大体上是相同的，这里OS是Ubuntu16.04，目标CUDA版本是10.1

准备条件

在此之前Ubuntu需要安装好Nvidia对应的显卡驱动
当然，gcc编译环境也是必要的

必要检查

检查是否GPU支持CUDA

lspci | grep -i nvidia

如果没有也没有，可能存在问题。 可以执行update-pciids再试一下。

下面是我机器的output。显卡不是太好，勉强用一下。

检查linux版本是否支持CUDA

cat /etc/lsb_release

检查一下你的版本是否是Ubuntu 16/18的版本

检查gcc版本

gcc --version

安装CUDA TOOLKIT

上述检查都做完后，我们开始正式安装吧。

首先要要去下载安装包。 有两种方式，一是去NVIDIA官网上下载。

还有一种方式是通过Package Manager.

这里我们通过Package Manager安装

通过PackageManager安装

Step 1. 下载repo包

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_10.1.243-1_amd64.deb

Step 2. 安装repo包

dpkg -i cuda-repo-ubuntu1804_10.1.105-1_amd64.deb

Step 3. 获取GPG key

sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub

Step 4. update

apt-get update

Step 5.安装

apt install cuda

引用文章：https://www.osetc.com/en/how-to-install-cuda-on-ubuntu-18-04-16-04-linux.html