blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-08-27 23:58:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
XXXXRT666 2025-05-01 03:47:02 +01:00
parent c14575ffec
commit 4c4a84f63a
2 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Docker/setup.sh
View File

@ -48,13 +48,9 @@ fi
source "$HOME/anaconda3/etc/profile.d/conda.sh" source "$HOME/anaconda3/etc/profile.d/conda.sh"
echo "CUDA_VERSION: $CUDA_VERSION" if [ "$CUDA_VERSION" = 12.8 ]; then
if [ "$CUDA_VERSION" = 128 ]; then
echo 1111111
pip install torch torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/cu128 pip install torch torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/cu128
elif [ "$CUDA_VERSION" = 124 ]; then elif [ "$CUDA_VERSION" = 12.4 ]; then
echo 2222222
pip install torch==2.5.1 torchaudio==2.5.1 --no-cache-dir --index-url https://download.pytorch.org/whl/cu124 pip install torch==2.5.1 torchaudio==2.5.1 --no-cache-dir --index-url https://download.pytorch.org/whl/cu124
fi fi

10
install.sh
View File

@ -69,11 +69,11 @@ while [[ $# -gt 0 ]]; do
--device) --device)
case "$2" in case "$2" in
CU124) CU124)
CUDA_VERSION=124 CUDA=124
USE_CUDA=true USE_CUDA=true
;; ;;
CU128) CU128)
CUDA_VERSION=128 CUDA=128
USE_CUDA=true USE_CUDA=true
;; ;;
ROCM) ROCM)
@ -228,9 +228,11 @@ fi
if [ "$USE_CUDA" = true ] && [ "$WORKFLOW" = false ]; then if [ "$USE_CUDA" = true ] && [ "$WORKFLOW" = false ]; then
echo "Installing PyTorch with CUDA support..." echo "Installing PyTorch with CUDA support..."
if [ "$CUDA_VERSION" = 128 ]; then if [ "$CUDA" = 128 ]; then
echo 11111
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_VERSION" = 124 ]; then elif [ "$CUDA" = 124 ]; then
echo 22222
pip install torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124 pip install torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124
fi fi
elif [ "$USE_ROCM" = true ] && [ "$WORKFLOW" = false ]; then elif [ "$USE_ROCM" = true ] && [ "$WORKFLOW" = false ]; then