diff --git a/.github/workflows/build_on_pr.yml b/.github/workflows/build_on_pr.yml index 308aebe8c..e84240fa5 100644 --- a/.github/workflows/build_on_pr.yml +++ b/.github/workflows/build_on_pr.yml @@ -89,7 +89,7 @@ jobs: if: needs.detect.outputs.anyLibraryFileChanged == 'true' runs-on: gpu-h20-10 container: - image: hpcaitech/pytorch-cuda:2.2.2-12.1.0 + image: image-cloud.luchentech.com/hpcaitech/pytorch-cuda:2.2.2-12.1.0 options: --gpus all --rm -v /dev/shm -v /data/scratch:/data/scratch timeout-minutes: 90 defaults: