From 3ea1472b0303d41375ae1ebb2042550293c2feaa Mon Sep 17 00:00:00 2001 From: Przemek Tredak Date: Mon, 27 Oct 2025 09:32:54 -0700 Subject: [PATCH] Change the pyTorch installation to CUDA 13 in Build All GitHub action to match the version in the JAX container Signed-off-by: Przemek Tredak --- .github/workflows/build.yml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index f40b281895..42c5f0342e 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -83,7 +83,9 @@ jobs: options: --user root steps: - name: 'Dependencies' - run: pip install torch pybind11[global] einops onnxscript + run: | + pip install pybind11[global] einops onnxscript + pip install torch --index-url https://download.pytorch.org/whl/cu130 - name: 'Checkout' uses: actions/checkout@v3 with: