Ubuntu 安装 cuda

下载

去 https://developer.nvidia.com/cuda-toolkit-archive 下载需要版本的cuda。下载的文件不只有cuda，还有对应的驱动。如果已经安装完驱动，可以在安装过程中选择只安装cuda。如果要安装驱动的话先安装适用于发行版的驱动和cuda，然后在需要的cuda，cuda可以存在多个，通过软连接切换。

# 获取 & 安装
wget https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run

安装

安装驱动的话会出现各种错误，除了/var/log/cuda-installer.log日志文件，/var/log/nvidia-installer.log中包含了更多可用的信息。安装驱动的时候如果提示gcc不兼容，你可能得切换gcc版本；如果安装cuda时提示gcc不兼容，直接使用--override选项忽略。
下面是切换gcc版本的命令

# 验证GCC版本，还是刚才那个网页，下载链接后面对应文档，在文档里能看到gcc版本
https://developer.nvidia.com/cuda-toolkit-archive

# 添加软件源，删除软件源的话用add-apt-repository -r
sudo add-apt-repository ppa:ubuntu-toolchain-r/test
# ubuntu22.04
sudo add-apt-repository deb [arch=amd64] http://archive.ubuntu.com/ubuntu focal main universe
sudo apt update

sudo apt-get install gcc-11.0
sudo apt-get install g++-11.0
# gcc 8
apt -y install gcc-8 g++-8

# 设置优先级
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-11 100
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-11 100
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-8 100
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-8 100
# 切换版本
sudo update-alternatives --config gcc
sudo update-alternatives --config g++

如果提示需要关闭图形界面：

# 设置使用命令行启动，重启生效
sudo systemctl set-default multi-user.target 
# 设置图形界面启动 
sudo systemctl set-default graphical.target 

# 从命令行启动之后可以使用xstart命令继续启动图形界面，ctrl+alt+<f1/f2>是图形界面，ctrl+alt+<f3-f6>是命令行界面

3. env

export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"