在使用GPU进行神经网络的训练或者进一步操作时,会遇到TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.这样的问题。
原因:
在我们想把 GPU tensor 转换成 Numpy 变量的时候,需要先将 tensor 转换到 CPU 去,因为 Numpy 是 CPU-only 的。
解决:
一般来说需要在你转换的变量后面加上.cpu()
比如我的: