网上各种解决方法,但是我都试了不可以,我实验发现如果不采用gpu环境的pytorch程序无报错,采用使用gpu的pytoch程序报错,采用gpu的tensroflow和keras不报错。这就让我很疑惑,为什么只有采用gpu的pytorch程序才会报错,又联想到windows10有自动更新系统的毛病,所以我怀疑是自动更新把我配的环境搞坏了。所以采取了一下自查行为:
1.查看自己安装torch和torchvision的版本
我的环境是windows10,python版本是3.6.7,笔记本显卡是GTX950m,torchvision版本是cu92/torchvision-0.6.0%2Bcu92-cp36-cp36m-win_amd64.whl,torch版本是cu92/torch-1.4.0%2Bcu92-cp36-cp36m-win_amd64.whl。这个版本是之前按Pytorch官网配置的,运行一直没有问题,从cu92可以看出,cuda环境是cuda==9.2。然后坑就在这里,我们接着看。
2.查看自己配置的cuda版本
通过在终端输入nvcc --version查看cuda配置的版本,发现是之前配置的9.2.
有意思的地方来了,我又通过