mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-08-24 22:05:59 +08:00
Update install.sh
This commit is contained in:
parent
c1e82e9b94
commit
cfefe6febf
@ -252,14 +252,14 @@ if [ "$USE_CUDA" = true ] && [ "$WORKFLOW" = false ]; then
|
|||||||
if [ "$CUDA" = 128 ]; then
|
if [ "$CUDA" = 128 ]; then
|
||||||
pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu128
|
pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu128
|
||||||
elif [ "$CUDA" = 124 ]; then
|
elif [ "$CUDA" = 124 ]; then
|
||||||
pip install torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124
|
pip install torch==2.6 torchaudio --index-url https://download.pytorch.org/whl/cu124
|
||||||
fi
|
fi
|
||||||
elif [ "$USE_ROCM" = true ] && [ "$WORKFLOW" = false ]; then
|
elif [ "$USE_ROCM" = true ] && [ "$WORKFLOW" = false ]; then
|
||||||
echo "Installing PyTorch with ROCm support..."
|
echo "Installing PyTorch with ROCm support..."
|
||||||
pip install torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/rocm6.2
|
pip install torch==2.6 torchaudio --index-url https://download.pytorch.org/whl/rocm6.2
|
||||||
elif [ "$USE_CPU" = true ] && [ "$WORKFLOW" = false ]; then
|
elif [ "$USE_CPU" = true ] && [ "$WORKFLOW" = false ]; then
|
||||||
echo "Installing PyTorch for CPU..."
|
echo "Installing PyTorch for CPU..."
|
||||||
pip install torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cpu
|
pip install torch==2.6 torchaudio --index-url https://download.pytorch.org/whl/cpu
|
||||||
elif [ "$WORKFLOW" = false ]; then
|
elif [ "$WORKFLOW" = false ]; then
|
||||||
echo "Unknown Err"
|
echo "Unknown Err"
|
||||||
exit 1
|
exit 1
|
||||||
|
Loading…
x
Reference in New Issue
Block a user