changed ROCm pip index URL from 5.2 to 5.1.1, because it's what worked for me desu

This commit is contained in:
mrq 2023-02-06 22:52:40 -06:00
parent 5d76d47a49
commit 6475045f87

View File

@ -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