Update test_cuda.yml

Signed-off-by: Jinzhe Zeng <[email protected]>

.github/workflows/test_cuda.yml

@@ -15,11 +15,13 @@ jobs:
       options: --gpus all
     if: github.repository_owner == 'deepmodeling' && github.event.label.name == 'Test CUDA' || github.event_name == 'workflow_dispatch'
     steps:
+    - name: Make sudo work
+      run: apt-get update && apt-get install -y sudo
     - uses: actions/checkout@v4
-    - uses: actions/setup-python@v4
-      with:
-        python-version: '3.11'
-        cache: 'pip'
+    # - uses: actions/setup-python@v4
+    #   with:
+    #     python-version: '3.11'
+    #     cache: 'pip'
     - name: Setup MPI
       uses: mpi4py/setup-mpi@v1
       with:
@@ -30,6 +32,8 @@ jobs:
          && sudo dpkg -i cuda-keyring_1.0-1_all.deb \
          && sudo apt-get update \
          && sudo apt-get -y install cuda-11-8 libcudnn8=8.9.5.*-1+cuda11.8
+    - name: Set PyPI mirror for Aliyun cloud machine
+      run: python -m pip config --user set global.index-url http://mirrors.cloud.aliyuncs.com/pypi/simple/
     - run: python -m pip install -U "pip>=21.3.1,!=23.0.0"
     - run: pip install -v -e .[gpu,test,lmp,cu11] "ase @ https://github.com/rosswhitfield/ase/archive/edd03571aff6944b77b4a4b055239f3c3e4eeb66.zip"
       env: