From 4bcb83782afafbb86dfde67d6151eb1631855e08 Mon Sep 17 00:00:00 2001 From: Jerry Zhang Date: Fri, 22 Nov 2024 11:26:02 -0800 Subject: [PATCH] Update nightly job to use 12.4 since 12.1 is deprecated https://github.com/pytorch/ao/pull/1278#issuecomment-2492933680 --- .github/workflows/regression_test.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/regression_test.yml b/.github/workflows/regression_test.yml index f1104bf66a..d9649b7f7e 100644 --- a/.github/workflows/regression_test.yml +++ b/.github/workflows/regression_test.yml @@ -25,9 +25,9 @@ jobs: include: - name: CUDA Nightly runs-on: linux.g5.12xlarge.nvidia.gpu - torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121' + torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu124' gpu-arch-type: "cuda" - gpu-arch-version: "12.1" + gpu-arch-version: "12.4" - name: CPU Nightly runs-on: linux.4xlarge torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cpu'