×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

yolov7移植推理调用torch.load提示找不到cuda设备 已完结 hzqsxxymh2023-03-25 13:34:15 回复 6 查看 技术答疑
yolov7移植推理调用torch.load提示找不到cuda设备
分享到:

仿照yolov5官方示例移植yolov7,在pytorch_yolov7_inference.py一开始调用加载原pt的load_dict函数后报错,提示找不到GPU设备,可以添加map_location=‘cpu’。添加后能正常进行推理,但为什么和寒武纪仓库提供的pytorch_yolov5_inference.py写的一样,运行yolov5是正常的,但yolov7就报错。

版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭