本文旨在帮助Pytorch使用者快速上手使用寒武纪MLU。以代码块为主，文字尽可能简洁，许多部分对标NVIDIA CUDA。不正确的地方请留言更正。本文不定期更新。 文章目录 前言Cambricon PyTorch的Python包torch_mlu导入将模型加载到MLU上model.to('mlu')定义损失函数，然后将其拷贝至MLU将数据从CPU拷贝到MLU设备以mnist.py为

支持情况 Pytorch 只支持 1.6.0 导出。 如何生成ONNX？ 解析失败 包括：Parse 过程转换失败node index，不支持的模型输入tensor name，不支持的控制流子图输入tensor name。 提取支持的子图