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()
喜欢的 留个 关注 、 加 点赞 哦 ~
相关推荐
IT：前端进阶技术路线图(初级→中级→高级)之初级(研发工具/HTML/CSS/JS/浏览器)/中级(研发链路/工程化/库/框架/性能优化/工作原理)/高级(搭建/中后台/体验管理等)之详细攻略
【历史上的今天】7 月 10 日：iOS App Store 问世；台积电创始人出生；苹果手机越狱的起源
YOLOv4 deepsort pytorch实现【代码资源已上传】
android界面布局----TableLayout（表单布局）
OpenCV每日函数 对象追踪模块 使用增强相关系数 (ECC) 最大化的图像配准
JetBrains IntelliJ IDEA 2021 for Mac(Java开发工具)
ios mysql注册登录界面_Vue+Vue Router+Axios+Webpack+Flask+MySQL实现简单的注册、登录验证功能...
大数据告诉你——垃圾围城，现行的垃圾分类制度，作用到底能有多大？