From 7f3a357b44fb4bb42a0c26e44e040ac644b6da10 Mon Sep 17 00:00:00 2001 From: Casper Date: Tue, 23 Jul 2024 18:45:52 +0200 Subject: [PATCH] pin torch==2.3.1 (#554) --- .github/workflows/build.yaml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 6b7c8f7e..a8adcede 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -91,7 +91,7 @@ jobs: # Install torch $cudaVersion = $env:CUDA_VERSION.Replace('.', '') $cudaVersionPytorch = $cudaVersion.Substring(0, $cudaVersion.Length - 1) - if ([int]$cudaVersionPytorch -gt 118) { $pytorchVersion = "torch==2.2.0" } else {$pytorchVersion = "torch==2.0.1"} + $pytorchVersion = "torch==2.3.1" python -m pip install --upgrade --no-cache-dir $pytorchVersion+cu$cudaVersionPytorch --index-url https://download.pytorch.org/whl/cu$cudaVersionPytorch python -m pip install build setuptools wheel ninja requests @@ -210,9 +210,9 @@ jobs: python -m pip install --upgrade build setuptools wheel requests if [[ "${{ matrix.rocm }}" == "5.7.1" ]]; then - python -m pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/rocm5.7 + python -m pip install torch==2.3.1 --index-url https://download.pytorch.org/whl/rocm5.7 elif [[ "${{ matrix.rocm }}" == "5.6.1" ]]; then - python -m pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/rocm5.6 + python -m pip install torch==2.3.1 --index-url https://download.pytorch.org/whl/rocm5.6 else echo Unknown rocm version for python install exit 1