11 lines
394 B
Bash
11 lines
394 B
Bash
|
python -m venv tortoise-venv
|
||
|
source ./tortoise-venv/bin/activate
|
||
|
python -m pip install --upgrade pip
|
||
|
# ROCM
|
||
|
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
|
||
|
# CUDA
|
||
|
# pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
|
||
|
python -m pip install -r ./requirements.txt
|
||
|
python setup.py install
|
||
|
deactivate
|