From 0669d1d5c5fc97202c22308c4cf6afcfc0cd8c58 Mon Sep 17 00:00:00 2001 From: cpuhrsch Date: Tue, 4 Jun 2024 15:58:19 -0400 Subject: [PATCH] Pin CUDA nightly to mitigate regression (#322) Mitigates https://github.com/pytorch/pytorch/issues/127924 --- .github/workflows/regression_test.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/regression_test.yml b/.github/workflows/regression_test.yml index 971515747e..ae76328517 100644 --- a/.github/workflows/regression_test.yml +++ b/.github/workflows/regression_test.yml @@ -33,7 +33,7 @@ jobs: gpu-arch-version: "12.1" - 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==2.4.0.dev20240603+cu121 --index-url https://download.pytorch.org/whl/nightly/cu121' gpu-arch-type: "cuda" gpu-arch-version: "12.1" - name: CPU 2.2.2