lightmare
/
tortoise-tts
forked from mrq/tortoise-tts
1
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

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

main
mrq 2023-02-06 22:52:40 +07:00
parent 6515d3b6de
commit 66cc6e2791
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
setup-rocm.sh
Unescape Escape 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