Fix dependency installation

casper-hansen · Feb 14, 2024 · 952969f · 952969f
1 parent f3581d6
commit 952969f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -209,9 +209,9 @@ jobs:
  python -m pip install --upgrade build setuptools wheel
 
  if [[ "${{ matrix.rocm }}" == "5.7.1" ]]; then
- python -m torch==2.2.0+rocm57 --index-url https://download.pytorch.org/whl/rocm5.7
+ python -m pip install torch==2.2.0+rocm57 --index-url https://download.pytorch.org/whl/rocm5.7
  elif [[ "${{ matrix.rocm }}" == "5.6.1" ]]; then
- python -m torch==2.2.0+rocm56 --index-url https://download.pytorch.org/whl/rocm5.6
+ python -m pip install torch==2.2.0+rocm56 --index-url https://download.pytorch.org/whl/rocm5.6
  else
  echo Unknown rocm version for python install
  exit 1