NPU (HUAWEI Ascend)¶
Please refer to the building documentation of MMCV to install MMCV on NPU devices
Here we use 4 NPUs on your computer to train the model with the following command:
bash tools/dist_train.sh configs/deeplabv3/deeplabv3_r50-d8_4xb2-40k_cityscapes-512x1024.py 4
Also, you can use only one NPU to train the model with the following command:
python tools/train.py configs/deeplabv3/deeplabv3_r50-d8_4xb2-40k_cityscapes-512x1024.py
If not specially marked, the results on NPU with amp are the basically same as those on the GPU with FP32.
All above models are provided by Huawei Ascend group.