TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory ..

在pytorch结合cuda下，直接使用

output.data.numpy()

会报标题错误：

TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

解决办法：

output.data.cpu().numpy()

把CUDA tensor格式的数据改成numpy时，需要先将其转换成cpu float-tensor随后再转到numpy格式。 numpy不能读取CUDA tensor 需要将它转化为 CPU tensor。

__EOF__

本文作者：龙雪的博客
本文链接：https://www.cnblogs.com/sakura-yuki/p/13341462.html
关于博主：评论和私信会在第一时间回复。或者直接私信我。
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！