ai-voice-cloning-fork/setup-rocm.sh
2023-02-19 21:41:51 +00:00

15 lines
479 B
Bash
Executable File

#!/bin/bash
git submodule init
git submodule update
python -m venv venv
source ./venv/bin/activate
python -m pip install --upgrade pip
# ROCM
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 ./dlas/requirements.txt
python -m pip install -r ./tortoise-tts/requirements.txt
python -m pip install -r ./requirements.txt
python -m pip install -e ./tortoise-tts/
deactivate