ai-voice-cloning/setup-rocm.sh

22 lines
799 B
Bash
Raw Permalink Normal View History

2023-02-18 02:46:44 +00:00
#!/bin/bash
# get local dependencies
2023-02-19 21:41:51 +00:00
git submodule init
git submodule update --remote
# setup venv
python3 -m venv venv
2023-02-17 00:08:27 +00:00
source ./venv/bin/activate
python3 -m pip install --upgrade pip # just to be safe
2023-02-17 00:08:27 +00:00
# ROCM
pip3 install torch==1.13.1 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2 # 5.4.2 doesn't work for me desu
# install requirements
python3 -m pip install -r ./modules/tortoise-tts/requirements.txt # install TorToiSe requirements
python3 -m pip install -e ./modules/tortoise-tts/ # install TorToiSe
python3 -m pip install -r ./modules/dlas/requirements.txt # instal DLAS requirements
python3 -m pip install -e ./modules/dlas/ # install DLAS
python3 -m pip install -r ./requirements.txt # install local requirements
rm *.bat
./setup-rocm-bnb.sh
deactivate