CUDA+CUDNN+torch+torchvision安装

弄了好久,终于弄好了!!!
CUDA+CUDNN+torch+torchvision安装_第1张图片

原因:其实之前我是已经配置好pytorch的相关环境的了。但是这段时间,在跑GNN相关论文中的代码时,发现代码中的某个函数要求torch必须得是1.8+
而我之前安装的是torch1.1+。但是更加棘手的是,我电脑上原本自带的驱动支持的最高版本是1.2+!所以需要先更新驱动,然后按照下面的顺序装
0、更新驱动
1、CUDA 11.1
2、CUDNN 8.2.0
3、torch 1.8.1
4、torchvision 0.9.1
5、torchaudio 0.8.1

第0步,这里尝试了很多中方法,DDU卸载、签名、关闭Windows Update服务、在安全模式下卸载后重装等等很多种方法,其实只需要在满足你需要的最低版本要求上,多尝试几个不同的显卡驱动安装就ok。甚至不需要断网、不需要关闭杀毒软件。。。
因为最近我还有课程设计需要做,具体的细节就不展开了。
如果需要的话,评论区回复我,我看到后一定及时回复。

第1和第2步,CUDA+CUDNN安装参考链接:

史上最详细的Pytorch+CUDA+CUDNN的安装(GPU版)

第3和第4步,torch和torchvision安装参考链接:

Python安装torch(含torch和torchvision对应版本)

你可能感兴趣的:(microsoft)