tortoise-tts/setup-cuda.sh
Bikkies 64ae4bb563 Updated setup scripts to use cuda 11.8 and torch 2.0.0 to fix RTX 4090 compatibility
Added API to the generate function, so it can be called from other scripts
2023-02-11 19:46:26 +11:00

11 lines
447 B
Bash

python -m venv tortoise-venv
source ./tortoise-venv/bin/activate
python -m pip install --upgrade pip
# CUDA
pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu118
pip install --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cu118
pip install --pre torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118
python -m pip install -r ./requirements.txt
python setup.py install
deactivate