ubuntu20.04的cuda支持到11.1了,但是pytorch目前只支持到11.0,所以需要的童靴需要注意下版本对应。
1. cuda官网下载,新安装的20.04是没有gcc和cmake需要自行安装,sudo apt install cmake gcc即可
Alt+ctrl+F3 进入tty3,关闭GUI
$ sudo systemctl set-default multi-user
重启
sudo ./cuda****(你下载的cuda全名) 然后accept, 选择all
之后,一致没错即可,一般不会有问题,有问题查看报错的日志
重启GUI
$ sudo systemctl set-default graphical
这就成功了,然后把cudnn的文件拷贝到/usr/local,在当前目录tar -zxvf cudnn****即可。
最后一步,就是添加环境变量
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}