use nvidia image

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

.github/workflows/test_cuda.yml

@@ -11,12 +11,12 @@ jobs:
     runs-on: nvidia
     # https://github.com/deepmodeling/deepmd-kit/pull/2884#issuecomment-1744216845
     container:
-      image: ubuntu:22.04
+      image: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
       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 wget
+      run: apt-get update && apt-get install -y sudo
     - uses: actions/checkout@v4
     - uses: actions/setup-python@v4
       with:
@@ -32,6 +32,7 @@ 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
+      if: false  # skip as we use nvidia image
     - 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"