下载
去 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>是命令行界面
- env
export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"