- 更新显卡驱动
选择最新的驱动点击Apply Changes,安装结束后重启电脑。
$ nvidia-smi //查看是否安装成功
图形显示NVIDIA即为安装成功。
- 下载并安装CUDA
显卡驱动与CUDA版本对应关系:
Release Notes :: CUDA Toolkit Documentation
官网下载CUDA:
CUDA Toolkit Archive | NVIDIA Developer
下载结束后,关闭图形界面:
首先Ctrl+Alt+F3(如果不行就1-7尝试)进入命令行模式,根据系统版本不同,关闭图形界面的命令也不同:
Ubuntu20.04:
$ sudo systemctl stop gdm3 // 关闭图形界面
$ sudo systemctl start gdm3 // 打开图形界面
//如果上面的不行,可以试试下面的
//设置开机默认进入命令行
$ sudo systemctl set-default multi-user.target
$ sudo reboot
//设置开机默认进入图形界面
$ sudo systemctl set-default graphical.target
$ sudo reboot
Ubuntu18.04:
//设置开机默认进入命令行
$ sudo systemctl set-default multi-user.target
$ sudo reboot
//设置开机默认进入图形界面
$ sudo systemctl set-default graphical.target
$ sudo reboot
开始安装:
$ sudo sh cuda_11.3.0_465.19.01_linux.run
按照提示操作。输入“accept”同意安装,进入安装确认界面在driver那一行按enter去掉【x】中的x,不安装driver。
安装结束后恢复图形界面,reboot。
添加环境变量:
$ sudo gedit ~/.bashrc
//在最后添加:
export PATH=/usr/local/cuda-11.3/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH
$ source ~/.bashrc
测试CUDA是否安装成功:
$ nvcc -V //查看CUDA版本信息
$ cd NVIDIA_CUDA-11.3_Samples
$ make -j8 //编译CUDA Samples
$ ./bin/x86_64/linux/release/deviceQuery
最后输出结果pass代表成功。
- 下载并安装cudnn
https://developer.nvidia.com/rdp/cudnn-download
需要注册并加入开发者计划
$ tar -zxvf cudnn-11.3-linux-x64-v8.2.1.32.tgz
$ sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-11.3/lib64/
$ sudo cp cuda/include/cudnn.h /usr/local/cuda-11.3/include/
# 为所有用户设置读取权限
$ sudo chmod a+r /usr/local/cuda-11.3/include/cudnn.h
$ sudo chmod a+r /usr/local/cuda-11.3/lib64/libcudnn*