Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit 952969fd authored by Casper's avatar Casper
Browse files

Fix dependency installation

parent f3581d6a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
.github/workflows/build.yaml
View file @ 952969fd
...@@ -209,9 +209,9 @@ jobs: ...@@ -209,9 +209,9 @@ jobs:
python -m pip install --upgrade build setuptools wheel python -m pip install --upgrade build setuptools wheel
if [[ "${{ matrix.rocm }}" == "5.7.1" ]]; then 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 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 else
echo Unknown rocm version for python install echo Unknown rocm version for python install
exit 1 exit 1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment