forked from mrq/tortoise-tts
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
This commit is contained in:
parent
9bf1ea5b0a
commit
64ae4bb563
|
@ -1,7 +1,9 @@
|
|||
python -m venv tortoise-venv
|
||||
call .\tortoise-venv\Scripts\activate.bat
|
||||
python -m pip install --upgrade pip
|
||||
python -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
|
||||
python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu118
|
||||
python -m pip install --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cu118
|
||||
python -m 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
|
||||
|
|
|
@ -2,7 +2,9 @@ python -m venv tortoise-venv
|
|||
source ./tortoise-venv/bin/activate
|
||||
python -m pip install --upgrade pip
|
||||
# CUDA
|
||||
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
|
||||
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
|
||||
|
|
Loading…
Reference in New Issue
Block a user