TypeError: can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor ..

查了一些帖子都说改源码，不建议

编辑错误，上面已经提示怎么改了。

tensor.numpy()改为：
tensor.cpu().numpy()

版权声明：本文为m0_61102347原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。