企业网站建设知识求网站建设网站优化工作

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

重启安装完成