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

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

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

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

你可能感兴趣的:(numpy,python,深度学习)