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

最新推荐文章于 2025-09-04 17:23:30 发布

转载最新推荐文章于 2025-09-04 17:23:30 发布 · 2.6k 阅读

2 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/huangxiaofu/p/11454299.html

文章标签：

#python

本文介绍如何正确地将PyTorch中的CUDAtensor格式数据转换为numpy格式，避免因直接调用numpy()方法导致的错误。通过先将数据转移到CPU上，再进行转换，可以有效解决这一问题。

部署运行你感兴趣的模型镜像

predict=predict.data.numpy() 这一行报错
意思是:如果想把CUDA tensor格式的数据改成numpy时
，需要先将其转换成cpu float-tensor随后再转到numpy格式。 
numpy不能读取CUDA tensor 需要将它转化为 CPU tensor
将predict.data.numpy() 改为predict.data.cpu().numpy()即可

转 https://blog.youkuaiyun.com/qq_38410428/article/details/82973711

转载于:https://www.cnblogs.com/huangxiaofu/p/11454299.html

您可能感兴趣的与本文相关的镜像