diff --git a/.github/workflows/regression_test.yml b/.github/workflows/regression_test.yml index 3aee8dbfb..13cd4e2e7 100644 --- a/.github/workflows/regression_test.yml +++ b/.github/workflows/regression_test.yml @@ -35,7 +35,7 @@ jobs: gpu-arch-version: "12.1" - name: CUDA 2.4 runs-on: linux.g5.12xlarge.nvidia.gpu - torch-spec: 'torch==2.4.0' + torch-spec: 'torch==2.4.1' gpu-arch-type: "cuda" gpu-arch-version: "12.1" - name: CUDA Nightly (Oct 1)