Skip to content

GitLab

Unverified Commit 9477ea5c authored by Casper's avatar Casper Committed by GitHub
Browse files

Fix ROCm build (#342)

parent bcaa8a36
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
.github/workflows/build.yaml
View file @ 9477ea5c
......@@ -209,11 +209,9 @@ jobs:
python -m pip install --upgrade build setuptools wheel
if [[ "${{ matrix.rocm }}" == "5.7.1" ]]; then
echo "Using PyTorch nightly"
python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/rocm5.7
python -m pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/rocm5.7
elif [[ "${{ matrix.rocm }}" == "5.6.1" ]]; then
echo "Using PyTorch stable"
python -m pip install torch --index-url https://download.pytorch.org/whl/rocm5.6
python -m pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/rocm5.6
else
echo Unknown rocm version for python install
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