/question about yolo model train

[aa851182472]

Does the pre-trained model of the YOLO series, such as v5, v8, and v11, support training on Huawei Ascend Chip 910B2? I would greatly appreciate it if you could read and respond to this inquiry!

[UltralyticsAssistant]

👋 Hello @aa851182472, thank you for your interest in Ultralytics 🚀! This is an automated response — an Ultralytics engineer will also assist you here shortly.

We recommend a visit to the Docs for new users where you can find many Python and CLI usage examples and where many of the most common questions may already be answered.

If this is a 🐛 Bug Report, please provide a minimum reproducible example to help us debug it.

If this is a custom training ❓ Question, please provide as much information as possible, including:
-Your OS and environment (container vs. bare metal)
-Python, PyTorch, and CUDA/CUDNN versions (if applicable)
-The device SDK/driver versions you’re using
-Exact commands you ran and any logs/errors
-And verify you are following our Tips for Best Training Results ✅

Join the Ultralytics community where it suits you best. For real-time chat, head to Discord 🎧. Prefer in-depth discussions? Check out Discourse. Or dive into threads on our Subreddit to share knowledge with the community.

Upgrade

Upgrade to the latest ultralytics package including all requirements in a Python>=3.8 environment with PyTorch>=1.8 to verify your issue is not already resolved in the latest version:

pip install -U ultralytics

Environments

YOLO may be run in any of the following up-to-date verified environments (with all dependencies including CUDA/CUDNN, Python and PyTorch preinstalled):

• Notebooks with free GPU:
• Google Cloud Deep Learning VM. See GCP Quickstart Guide
• Amazon Deep Learning AMI. See AWS Quickstart Guide
• Docker Image. See Docker Quickstart Guide

Status

If this badge is green, all Ultralytics CI tests are currently passing. CI tests verify correct operation of all YOLO Modes and Tasks on macOS, Windows, and Ubuntu every 24 hours and on every commit.

[Y-T-G]

It's not supported. But there's a PR by a user:

#22270

[glenn-jocher]

Hi @aa851182472, a related Huawei Ascend update: native Ascend export, inference, and validation support has now shipped in Ultralytics. Models can be exported directly to .om format with format=ascend, and Ultralytics Platform now provides managed exports for Ascend310P1, Ascend310P3, Ascend310B1, and Ascend310B4.

The complete workflow is documented in the Huawei Ascend integration guide.

This release does not add training support for Ascend 910B2. Ultralytics can select a single npu device through torch_npu, but fully validated Ascend training and multi-NPU HCCL DDP remain unsupported.