×

签到

分享到微信

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

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

【PyTorch 2.4】QAT量化时的训练是在cuda上还是寒武纪单板上? 蓁蓁2025-08-13 16:42:10 回复 3 查看 技术答疑 使用求助 经验交流
【PyTorch 2.4】QAT量化时的训练是在cuda上还是寒武纪单板上?
分享到:

【寒武纪硬件产品型号】

mlu370


【使用操作系统】
ubuntu


【使用驱动版本】
v4.20.6


【出错信息】在pytorch框架里对单板进行QAT量化时需要训练,请问训练时device是cuda还是370单版吗?谢谢。



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