cuda11+pytorch安装

1 .查看显卡的驱动版本

帮助-> 系统信息

可知我的是cuda11的驱动

然后就去官网找对应的安装包下载

我选择了本地安装

精简安装

 

 

安装完成后测试看看是否安装成功

控制台输入

下载cuDNN

https://developer.nvidia.com/rdp/cudnn-archive

到这网站去寻找对应的版本

安装cuDNN

解压后得到三个文件夹，把这三个文件夹中的内容分别放到之前cuda对应的文件夹中。

2 安装pytorch

pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple/

使用这个源会比较快

打开环境测试