Added a CPU setup script (based on the existing CUDA and ROCm setup scripts)

This commit is contained in:
Yacine Cheikhrouhou 2023-11-13 10:02:18 +01:00
parent 94f88886b0
commit 654aa69f12

22
setup-cpu.sh Executable file
View File

@ -0,0 +1,22 @@
#!/bin/bash
# get local dependencies
git submodule init
git submodule update --remote
# setup venv
python3 -m venv venv
source ./venv/bin/activate
python3 -m pip install --upgrade pip # just to be safe
# CUDA
#pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
# CPU
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
# 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, last, because whisperx will break a dependency here
python3 -m pip install -e ./modules/dlas/ # install DLAS
python3 -m pip install -r ./requirements.txt # install local requirements
rm *.bat
deactivate