Skip to content

Commit 5a551a2

Browse files
authored
[CI] Update CANN image address (#46)
Due to the replacement of the Ascend machine equipment, the dockerhub website image cannot be used now. It has been changed to the Huawei Cloud image address.
1 parent 9deda13 commit 5a551a2

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

.github/workflows/ascend_npu_test.yml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,10 @@ on:
5151
required: true
5252
type: choice
5353
options:
54-
- ascendai/cann:7.0.1-910b-ubuntu22.04-py3.8
55-
- ascendai/cann:8.0.0-910b-ubuntu22.04-py3.10
56-
- ascendai/cann:latest
57-
default: "ascendai/cann:latest"
54+
- swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:8.1.rc1-910b-ubuntu22.04-py3.11
55+
- swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:8.3.rc2-910b-ubuntu22.04-py3.11
56+
- swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:latest
57+
default: "swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:latest"
5858
description: "The docker image which will be loaded"
5959

6060
# Only cancel the previous runs when triggered by a pull_request event
@@ -83,7 +83,7 @@ jobs:
8383
id: set-env
8484
run: |
8585
echo "runner=${{ github.event.inputs.runner || 'linux-arm64-npu-1' }}" >> $GITHUB_OUTPUT
86-
echo "image=${{ github.event.inputs.image || 'ascendai/cann:latest' }}" >> $GITHUB_OUTPUT
86+
echo "image=${{ github.event.inputs.image || 'swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:latest' }}" >> $GITHUB_OUTPUT
8787
8888
# TODO(shink): List ghstack PR's ref
8989
- name: List ref to the PyTorch branch

0 commit comments

Comments
 (0)