在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