1. 驱动安装
1.1. 查看适合的版本
apt-get update
ubuntu-drivers devices
输出结果:
1.2. 安装合适的驱动版本
根据上面输出的内容
apt-get install nvidia-driver-545
完成后重启
reboot
查看新的驱动
nvidia-smi
2. 安装/升级cuda
在nvidia-smi中显示的CUDA Version
并非真正自己的cuda版本,而是当前驱动的最高可支持cuda版本,因此如果需要升级cuda需要额外操作。
2.1. 查看当前cuda版本
nvcc -V
2.2. 官网安装新的cuda
去官网找到自己要的版本
- 查看Architecture
uname -m
- 查看操作系统版本
lsb_release -a
- 官网给出的cuda安装命令
注:--fix-missing
是我自己加上的
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-keyring_1.0-1_all.deb
sudo dpkg -i cuda-keyring_1.0-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda --fix-missing
重启安装完成