From 5523a72adcd05175100022d35ef72a99e00d1278 Mon Sep 17 00:00:00 2001
From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com>
Date: Wed, 2 Apr 2025 03:40:20 +0100
Subject: [PATCH] update pytorch version and colab

---
 colab_webui.ipynb | 5 +----
 install.sh        | 6 +++---
 2 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/colab_webui.ipynb b/colab_webui.ipynb
index 226cc21..5faee51 100644
--- a/colab_webui.ipynb
+++ b/colab_webui.ipynb
@@ -33,11 +33,8 @@
         "condacolab.install_from_url(\"https://repo.anaconda.com/miniconda/Miniconda3-py39_23.11.0-2-Linux-x86_64.sh\")\n",
         "%cd -q /content\n",
         "!git clone https://github.com/RVC-Boss/GPT-SoVITS\n",
-        "!conda install -y -q -c pytorch -c nvidia cudatoolkit\n",
         "%cd -q /content/GPT-SoVITS\n",
-        "!conda install -y -q -c conda-forge gcc gxx ffmpeg cmake -c pytorch -c nvidia\n",
-        "!/usr/local/bin/pip install -r extra-req.txt --no-deps\n",
-        "!/usr/local/bin/pip install -r requirements.txt"
+        "!bash install.sh"
       ]
     },
     {
diff --git a/install.sh b/install.sh
index 66a3a49..2f93e9f 100644
--- a/install.sh
+++ b/install.sh
@@ -48,13 +48,13 @@ fi
 
 if [ "$USE_CUDA" = true ]; then
     echo "Installing PyTorch with CUDA support..."
-    conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 pytorch-cuda=11.8 -c pytorch -c nvidia
+    pip install torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124
 elif [ "$USE_ROCM" = true ]; then
     echo "Installing PyTorch with ROCm support..."
-    pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/rocm6.2
+    pip install torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/rocm6.2
 else
     echo "Installing PyTorch for CPU..."
-    conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 cpuonly -c pytorch
+    pip install torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cpu
 fi
 
 echo "Installing Python dependencies from requirements.txt..."