一、安装显卡驱动
1. 终端中输入以下命令获取显卡和驱动信息:
ubuntu-drivers devices
以我自己的机器为例,显示结果如下:
2. 我这里选择recommended的驱动版本进行安装:
sudo apt install nvidia-driver-515
安装完成后输入nvidia-smi进行测试,若显示如下就代表安装成功。
二、安装CUDA
1. 在官网(https://developer.nvidia.com/cuda-toolkit-archive)下载cuda,版本根据自己需求选择即可,但是不要超过显卡支持的最高版本。
2. 然后选择runfile的方式安装(因为只用输入两行命令,懒…)
3. 下载完成后终端输入sudo sh cuda_11.6.2_510.47.03_linux.run,按照红框标注的进行选择即可:
4. 这里有一点需要注意一下,因为之前已经安装过显卡驱动了,所以这里需