【bug】TypeError:can’t convert cuda:0 device type tensor to numpy.

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转换