切换版块
×
基础软件平台
PyTorch开发
TensorFlow开发
编解码及SDK开发
BANG语言与计算库
开发工具链
MagicMind开发
软件栈百科
云平台集成
硬件产品专区
MLU370系列AI加速卡
MLU270系列AI加速卡
MLU220系列AI加速产品
经验方案交流区
经验案例与实践分享
开发者服务
开发者活动
公告与版务
高校支持
《智能计算系统》
签到
版块
社区
文档
SDK下载
370系列
200系列
开发平台
官网首页
注册
登录
全部版块
基础软件平台
硬件产品专区
经验方案交流区
开发者服务
高校支持
发布新帖
登录/注册
LV.1
hzqsxxymh
24
积分
0
赞
2
帖子
8
回复
0
收藏
TA的动态
TA的帖子
TA的回复
yolov7移植推理调用torch.load提示找不到cuda设备
我的回复:#3hellokitty回复您好,使用原生pytorch模型进行magicmind模型迁移,是需要在cpu上加载模型权重的;建议添加map_location=‘cpu’另外,yolov7的magicmind移植,可以参考这个demo:https://gitee.com/cambricon/magicmind_cloud/tree/master/buildin/cv/detection/yolov7_pytorch展开但我看yolov5的例子可以在mlu上加载权重啊,但我yolov7照着写就是报错,这可能是什么原因导致的呢
0
yolov7移植推理调用torch.load提示找不到cuda设备
我的回复:#1hellokitty回复您好,请问您使用的是哪款寒武纪产品?“寒武纪仓库提供的pytorch_yolov5_inference.py”这是指哪个仓库的示例,请提供下链接展开https://gitee.com/cambricon/practices/tree/master/projects/pytorch_yolov5_inference
0
Github
开发平台
文档中心
新手必读
官方微信
版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:
京ICP备17003415号-1
关闭