YlGu
YlGu
> 你好，你用的torch是什么版本的，还是cuda版本的原因？ 我的版本是pytorch1.3.1 cudatoolkit 10.1.243
> 我是1.2.0 cuda 10.1 应该没问题的呀 开了个干净的虚拟环境，保持版本一致pytorch 1.2.0、torchvision 0.4，问题解决了，多谢~