TypeError: can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu()

发布时间:2023-12-19 08:00


Author :Horizon Max

编程技巧篇:各种操作小结

机器视觉篇:会变魔术 OpenCV

深度学习篇:简单入门 PyTorch

神经网络篇:经典网络模型

算法篇:再忙也别忘了 LeetCode


错误提示

在将 tensor 格式转换成 numpy 格式时出现报错:

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


错误原因

需要先将 tensor 转换到 CPU ,因为 Numpy 是 CPU-only


解决方案

使用 .cpu() 先进行转换

修改前:

predict.numpy()

修改后:

predict.cpu().numpy()

喜欢的 留个 关注 、 加 点赞 哦 ~



ItVuer - 免责声明 - 关于我们 - 联系我们

本网站信息来源于互联网,如有侵权请联系:561261067@qq.com

桂ICP备16001015号