打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
【寒武纪硬件产品型号】必填*:
MLU270,MLU220
【使用操作系统】必填*:
ubuntu 18.04
【使用驱动版本】必填*:
MLU270: v4.9.4
MLU220: v4.9.4
【使用AI框架】必填*:
Pytorch yolov5s
【操作步骤】选填:
1、系统上同时插上MLU270, MLU220
2、export CNRT_DEFAULT_DEVICE=1 (MLU270)
3、MLU270上生成270的离线模型并成功完成推理。
4、修改core_version:
ct.set_core_version('MLU220')
ct.save_as_cambricon('yolov5s_int8_1_4')
torch.set_grad_enabled(False)
ct.set_core_number(4)
trace_input = torch.randn(1, 3, 640, 640, dtype=torch.float).to(ct.mlu_device())
net = torch.jit.trace(quantized_net, trace_input, check_trace = False)
net(trace_input)
ct.save_as_cambricon("")
5、在MLU270上生成MLU220的模型报错。
【出错信息】必填*:
2022-07-04 20:05:50.382283: [cnrtWarning] [34391] [Card : 1] The device you run is not the same as the platform in kernel!
2022-07-04 20:05:50.382316: [cnrtWarning] [34391] [Card : 1] The device you run is MLU270, but the platform in kernel is MLU220!
【当前已做了哪些信息确认】选填:
设置core_version为MLU270可以生成离线模型并推理
请问还需要做什么修改吗?
热门帖子
精华帖子