From 890127fadfaedbe6d78515b2828241960612b395 Mon Sep 17 00:00:00 2001 From: Casper Date: Thu, 15 Feb 2024 19:05:05 +0100 Subject: [PATCH] Fix dependency (#343) --- .github/workflows/build.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 2e66d600..6b35f802 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -93,7 +93,7 @@ jobs: $cudaVersionPytorch = $cudaVersion.Substring(0, $cudaVersion.Length - 1) if ([int]$cudaVersionPytorch -gt 118) { $pytorchVersion = "torch==2.2.0" } else {$pytorchVersion = "torch==2.0.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 + python -m pip install build setuptools wheel ninja requests # Print version information python --version @@ -206,7 +206,7 @@ jobs: sudo apt-get update sudo apt-get install -y --no-install-recommends rocsparse-dev rocthrust-dev rocblas-dev hipblas-dev hipsparse-dev - python -m pip install --upgrade build setuptools wheel + 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