2023-02-18 02:46:44 +00:00
|
|
|
#!/bin/bash
|
2023-03-02 01:35:12 +00:00
|
|
|
# get local dependencies
|
2023-02-19 21:41:51 +00:00
|
|
|
git submodule init
|
2023-02-23 07:05:39 +00:00
|
|
|
git submodule update --remote
|
2023-03-02 01:35:12 +00:00
|
|
|
# setup venv
|
2023-02-21 20:20:52 +00:00
|
|
|
python3 -m venv venv
|
2023-02-17 00:08:27 +00:00
|
|
|
source ./venv/bin/activate
|
2023-03-02 01:35:12 +00:00
|
|
|
python3 -m pip install --upgrade pip # just to be safe
|
2023-02-17 00:08:27 +00:00
|
|
|
# ROCM
|
2023-03-02 01:35:12 +00:00
|
|
|
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.1.1 # 5.2 does not work for me desu
|
|
|
|
|
# install requirements
|
|
|
|
|
python3 -m pip install -r ./dlas/requirements.txt # instal DLAS requirements
|
|
|
|
|
python3 -m pip install -r ./tortoise-tts/requirements.txt # install TorToiSe requirements
|
|
|
|
|
python3 -m pip install -e ./tortoise-tts/ # install TorToiSe
|
|
|
|
|
python3 -m pip install -r ./requirements.txt # install local requirements
|
|
|
|
|
# swap to ROCm version of BitsAndBytes
|
|
|
|
|
pip3 uninstall bitsandbytes
|
|
|
|
|
pip3 install git+https://github.com/broncotc/bitsandbytes-rocm
|
|
|
|
|
|
|
|
|
|
deactivate
|
