【已解决TypeError: can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor
在使用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 转
·
在使用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()
比如我的：
为开发者提供学习成长、分享交流、生态实践、资源工具等服务，帮助开发者快速成长。
更多推荐
37
0- 0
已为社区贡献2条内容
所有评论(0)