diff --git a/.github/workflows/runner_ci.yml b/.github/workflows/runner_ci.yml index abb4175c..c72f0786 100644 --- a/.github/workflows/runner_ci.yml +++ b/.github/workflows/runner_ci.yml @@ -12,7 +12,7 @@ on: jobs: check-cuda: - runs-on: [gpumode-nvidia-arc] + runs-on: gpu-t4-4-core timeout-minutes: 10 container: image: nvidia/cuda:12.4.0-devel-ubuntu22.04 @@ -48,7 +48,7 @@ jobs: CUDA_VISIBLE_DEVICES: 0 check-pytorch: - runs-on: [gpumode-nvidia-arc] + runs-on: gpu-t4-4-core timeout-minutes: 10 container: image: nvidia/cuda:12.4.0-devel-ubuntu22.04