Update huawei-ascend-npu.yml

cosdt · Sep 4, 2024 · d5b0a77 · d5b0a77
1 parent ef40f4e
commit d5b0a77
Showing 1 changed file with 12 additions and 10 deletions.
diff --git a/.github/workflows/huawei-ascend-npu.yml b/.github/workflows/huawei-ascend-npu.yml
@@ -22,19 +22,21 @@ jobs:
     runs-on: [self-hosted, ascend, npu]
     container:
       image: ascendai/cann
-      ports:
-        - 80
-      options: --device /dev/davinci4
+      # ports:
+      #   - 80
+      volumes:
+        - /usr/local/dcmi:/usr/local/dcmi
+        - /usr/local/bin/npu-smi:/usr/local/bin/npu-smi 
+        - /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/
+        - /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info
+        - /etc/ascend_install.info:/etc/ascend_install.info   
+      options: --network host
+               --device /dev/davinci4
                --device /dev/davinci_manager
                --device /dev/devmm_svm
                --device /dev/hisi_hdc
-               --shm-size "20g"
-               -v /usr/local/dcmi:/usr/local/dcmi
-               -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi 
-               -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/
-               -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info
-               -v /etc/ascend_install.info:/etc/ascend_install.info      
-
+               --shm-size "20g"    
+
     steps:
     - uses: actions/checkout@v4