debug发现,是在方向传播那一块然后爆出这个错误
其实最终的解决办法是把pytorch卸载了之后再安装的
看了一下pytorch的版本,感觉自己的好像不对,但是平时其他时候用也没啥问题啊,神奇。。。建议大家都检查一下自己的pytorch版本和cuda版本是否对得上号
其实后来又遇见过这样的问题感觉大部分都是版本号和cuda不对号 或者torchvision(好像是这拼的)
可以去官网下载whl文件本地安装
注意看 cuda多少 python号 以及win还是linux系统 基本上就能解决上述问题