环境配置
python3.6+pytorch1.2.0+CUDA9.2+cuDNN7.1.4+cudatoolkit9.2+vs2019
一切看上去都那么正常…
跑程序的时候报了标题的错误,后悔没有截图纪念一下(当然也可以作死再换回去)
调试工具走起,一步步定位到loss.backward()这一操作,就在这崩的,我心态也跟着崩了…
找了好多方法,控制变量排除各种包的冲突numpy、matplotlib啊啥的,还有说相对路径改为绝对路径什么有的没的,但我数据集已经加载过了,明显不是这个问题。也要感谢numpy,虽然不是它的原因,我把numpy卸载了之后,我的pytorch也跟着卸载了,然后我去官网找链接下pytorch官网
刚想运行conda install pytorch==1.2.0 torchvision cudatoolkit=9.2
的时候,忍不住点开了这个救命链接直达
一直漫无目的地看着…直到这个表格!!!我是vs2019,pytorch至少要1.3.0才行!,由于我电脑也有CUDA10.1,所以conda install pytorch==1.3.0 torchvision cudatoolkit=10.1
最终配置:python3.6+pytorch1.3.0+CUDA10.1+cuDNN8.0.4+cudatoolkit10.1+vs2019
终于可以睡觉~~~了