a = torch.softmax(a, dim=1)
a = a.numpy()
在CPU上是正常运行的,然后用GPU的时候就出现了这个报错。
TypeError: can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
numpy不能直接读取CUDA tensor,需要将它转化为 CPU tensor。如果想把CUDA tensor格式的数据改成numpy,需要先将其转换成cpu float-tensor之后再转到numpy格式。
他已经告诉我们修改方法了,要先把 a 修改成 a.cpu()
a = a.cpu().numpy()
改成这个样子就好了!
修改过程中,第一次改的时候忘记加括号了,改成了:
a = a.cpu.numpy()
会出现新的报错,记得把括号加上!
'builtin_function_or_method' object has no attribute 'numpy'
还有问题可以参考这个文章看看:pytorch:tensor与numpy转换