From 6475045f870a2dcff28af880f0b60c850fd5b69c Mon Sep 17 00:00:00 2001 From: mrq Date: Mon, 6 Feb 2023 22:52:40 -0600 Subject: [PATCH] changed ROCm pip index URL from 5.2 to 5.1.1, because it's what worked for me desu --- setup-rocm.sh | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/setup-rocm.sh b/setup-rocm.sh index 9a54018..c128dba 100644 --- a/setup-rocm.sh +++ b/setup-rocm.sh @@ -2,9 +2,7 @@ python -m venv tortoise-venv source ./tortoise-venv/bin/activate python -m pip install --upgrade pip # ROCM -pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2 -# CUDA -# pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 +pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.1.1 # 5.2 does not work for me desu python -m pip install -r ./requirements.txt python setup.py install deactivate