我尝试了网上所有办法都没成功,反而一波未平一波又起,简单介绍下我最后的应对方法。
1. 配置cudnn环境,网上有很多教程,注意一下自己的版本对应就行。(必须配置完环境再进行下一步才不会出现torch引用不成功的问题)。此外,我把tensorflow也装了,建议大家也一起装上,减小变量。
2. 根据pytorch官网( https://pytorch.org/ )指令直接在cmd中进行下载安装,顺序如图:
我设置的环境命名为torch,首先进入环境中
我的指令是:conda install pytorch torchvision cudatoolkit=9.0 -c pytorch,我的这个指令不用再额外下载torchvision,但是有的博主的指令是两条,必须单独下载torchvision,大家注意下自己的版本就行。
此外,我这个是已经安装完的,所以没有显示进程(因为用的conda指令所以当时下载pytorch的时间有一点点久,我怕节外生枝就没有使用国内博主提供的便携下载途径),我的这个出现了一个警告,我没有理会,但是按照常理来说不会出现这个warning,目前没有心思解决这个问题,如果有出现同样问题的朋友可以探讨一下解决办法。
然后测试下是否成功:
没有出现问题。