Ubuntu 安装 CUDA

官方文档 | cuda-toolkit-archive

本文使用CUDA版本为 CUDA Toolkit 10.2 ，本文的Ubuntu版本为 Ubuntu 18.04 。

安装前建议关闭安全启动(Secure Boot)，否则可能会需要向安全启动器中写入什么，如果是双系统不知道会不会对主系统引导出现什么问题。

安装基础环境

安装gcc:

sudo apt-get install build-essential

安装linux头:

sudo apt-get install linux-headers-${uname -r}

安装前检查

在系统需求里检查设备是否符合安装需要。

# 验证显卡兼容
lspci | grep -i nvidia

# 验证Linux版本支持
uname -m && cat /etc/*release

# 验证gcc版本正确
gcc --version

正式安装

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

下载 cuda-ubuntu18.04.pin 完成后，检查一下文件内容，应类似如下:

Package: nsight-compute
Pin: origin *ubuntu.com*
Pin-Priority: -1

Package: nsight-systems
Pin: origin *ubuntu.com*
Pin-Priority: -1

Package: *
Pin: release l=NVIDIA CUDA
Pin-Priority: 600

安装完成后操作

设置环境变量

sudo gedit /etc/profile

添加如下:

export PATH=/usr/local/cuda-10.2/bin:/usr/local/cuda-10.2/NsightCompute-2019.1${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

设置完环境变量执行 source /etc/profile，之后重启系统。

测试

nvcc -V

正常显示版本即为安装成功。

(可选)安装可能会用到的第三方库

sudo apt-get install g++ freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev