TypeError: can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu()
发布时间：2023-01-28 11:30
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()
喜欢的 留个 关注 、 加 点赞 哦 ~
相关推荐
运用计算机思维可以解决什么问题,计算机思维是能否解决生活中遇到的难题？...
安装运行vue-element-admin的报错问题-解决办法
【比赛报告】biendata_2021搜狐校园文本匹配算法大赛_解决方案
向量化执行引擎框架 Gluten 宣布正式开源，并亮相 Spark 技术峰会
【浅墨著作】 OpenCV3编程入门 内容简介 勘误 配套源代码下载
Android 实现圆弧背景（Shape实现和自定义View）
fastai v1环境搭建：Win10 MX250 CUDA10.1 cuDNN Pytorch1.0.0 Fastai v1安(bi)装(keng)指南
python turtle绕原点旋转_中秋节，我们用Python代码做个五仁月饼吧
HashMap设计原理与实现（下篇）200行带你写自己的HashMap！！！