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:
parent
5d76d47a49
commit
6475045f87
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue
Block a user