Update huawei-ascend-npu.yml

cosdt · Sep 4, 2024 · 707ee25 · 707ee25
1 parent fa0451d
commit 707ee25
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/.github/workflows/huawei-ascend-npu.yml b/.github/workflows/huawei-ascend-npu.yml
@@ -42,6 +42,10 @@ jobs:
       run: |
         npu-smi info
         apt-get update
+        echo "export LD_LIBRARY_PATH=/usr/local/Ascend/driver/lib64/common:/usr/local/Ascend/driver/lib64/driver:${LD_LIBRARY_PATH}" >>~/.bashrc
+        echo "source /usr/local/Ascend/ascend-toolkit/set_env.sh" >>~/.bashrc
+        cat ~/.bashrc
+
         pip install torch==2.2.0 torchvision==0.17.0 torch_npu==2.2.0 numpy==1.26.4 cloudpickle tornado -i https://pypi.tuna.tsinghua.edu.cn/simple
         python -c "import torch; print('torch:', torch.__version__)"
         python -c "import torch,torch_npu; print('torch_npu:', torch.npu.is_available(),',version:', torch_npu.__version__)"