以下安装的版本
1、英伟达驱动:515.65.01
2、CUDA:11.0.2
3、CUDNN: 8.0.5.29
4、pytorch安装的方式,在官网选择命令行输入
conda install pytorch torchvision torchaudio cudatoolkit=11.3
下载英伟达驱动
(43条消息) Ubuntu20.04安装英伟达驱动步骤_Rootzzy1216的博客-优快云博客
下载cuda
1、在官网查看合适的版本
我此处下载11.0.2版本
2、按照下面教程下载