昇腾910B不支持paddleocr模型训练吗

[burth-1]

训练数据和模型参数加载place都是在npu上，但是训练起来npu的AICore就是没有使用起来，训练速度和使用cpu、训练差不多

[GreatV]

根据提供的文档和相关 Issue #13981，目前 PaddleOCR 对昇腾 910 (Ascend 910) 的支持存在一定的限制，且可能需要额外的配置工作。以下是相关信息和建议：

1. PaddleOCR 是否支持昇腾 910

在 Issue 中提到：

• PaddleOCR 2.6 的命令行帮助中没有明确的 NPU（如昇腾 910）选项。
• PaddleOCR 的运行环境支持需要 PaddlePaddle 框架的底层支持，而 PaddlePaddle 对昇腾 910 的支持依赖于 Huawei 的 CANN 驱动和相关适配。

虽然 PaddlePaddle 有部分对昇腾设备的支持，但具体到 PaddleOCR 的完整训练流程，目前可能尚未完全优化或适配。

---

2. 问题分析：为什么 AICore 不工作或表现类似 CPU

根据描述，训练时 AICore 没有被利用，速度与 CPU 类似。这可能由以下原因引起：

• 设备未正确初始化： 确保训练脚本中明确指定了 NPU 设备（如 place=paddle.set_device('npu')）。
• 驱动或环境不匹配： 检查是否安装了兼容的 CANN 驱动版本（如 CANN 7.0.1），并确保 Docker 镜像与硬件环境的适配。
• 模型未针对 NPU 优化： PaddleOCR 默认的模型和算子可能未针对昇腾 910 的架构进行优化，导致性能未能完全发挥。
• 框架支持不完整： 如果 PaddlePaddle 的 NPU 版本（如 paddle-npu）不支持某些关键算子，模型可能会回退到 CPU。

---

3. 建议解决方案

1. 检查环境配置：

   • 确保使用的是官方推荐的 Docker 镜像（如 registry.baidubce.com/device/paddle-npu:cann701-ubuntu20-aarch64-gcc84-py39）。
   • 检查 CANN 驱动和硬件环境是否正常工作，可参考华为昇腾的官方文档进行验证。

2. 明确设备设置：
   在训练代码中，确保使用以下代码段来指定 NPU：

   import paddle
   paddle.set_device('npu')

   确保 paddle 能够正确识别 NPU 设备。

3. 联系 PaddleOCR 社区：
   如果问题仍然存在，建议在 PaddleOCR Issues 页面提交详细的环境和日志信息，寻求开发者帮助。

4. 性能优化：
   如果模型未针对 NPU 优化，可以尝试：

   • 使用 PaddlePaddle 提供的更高版本（如有）。
   • 调整 PaddleOCR 的模型配置（如简化模型结构或调整 batch size）。

---

4. 参考链接

• PaddleOCR Issue: 是否支持昇腾 910
• PaddlePaddle NPU 支持文档
• 华为昇腾 CANN 官方文档

---

总结

PaddleOCR 对昇腾 910 的支持尚存在一定局限性，训练速度类似于 CPU 的问题可能是由于环境配置、模型优化或框架支持不完整导致的。建议检查环境配置、明确设备设置，并联系社区获得进一步支持。

---

Response generated by feifei-bot | chatgpt-4o-latest