diff --git a/.github/workflows/huawei-ascend-npu.yml b/.github/workflows/huawei-ascend-npu.yml
index d2b61144d342..966dea83ce5c 100644
--- 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__)"