以下安装的版本
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、按照下面教程下载
本文档详细记录了在Ubuntu20.04上安装英伟达驱动515.65.01、CUDA11.0.2、cuDNN8.0.5的过程,以及通过conda安装Pytorch、torchvision和torchaudio,特别提到了cudatoolkit=11.3的指定版本安装。同时,提供了卸载Pytorch的pip和conda方法。
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的博客-优快云博客
1、在官网查看合适的版本
我此处下载11.0.2版本
2、按照下面教程下载
1万+

被折叠的 条评论
为什么被折叠?