Update pytorch.yaml

using npu 7 instead of 1
Ascend · Oct 28, 2023 · f0a8f07 · f0a8f07
1 parent 7fbdbba
commit f0a8f07
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/pytorch.yaml b/.github/workflows/pytorch.yaml
@@ -51,7 +51,7 @@ jobs:
         - /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/
         - /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info
         - /root/codes/:/root/codes
-      options: --network host --device /dev/davinci1 --device /dev/davinci_manager --device /dev/devmm_svm --device /dev/hisi_hdc
+      options: --network host --device /dev/davinci7 --device /dev/davinci_manager --device /dev/devmm_svm --device /dev/hisi_hdc
     steps:
       - name: Set environment
         run: