From 768040a9d4f2182fdfa97280a9cb88e857707759 Mon Sep 17 00:00:00 2001
From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com>
Date: Thu, 22 May 2025 17:33:30 +0300
Subject: [PATCH] Update miniconda_install.sh

---
 Docker/miniconda_install.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Docker/miniconda_install.sh b/Docker/miniconda_install.sh
index 36ca5c53..001a2a46 100644
--- a/Docker/miniconda_install.sh
+++ b/Docker/miniconda_install.sh
@@ -55,8 +55,8 @@ source "$HOME/miniconda3/etc/profile.d/conda.sh"
 
 if [ "$CUDA_VERSION" = "12.8" ]; then
     "$HOME/miniconda3/bin/pip" install torch torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/cu128
-elif [ "$CUDA_VERSION" = "12.4" ]; then
-    "$HOME/miniconda3/bin/pip" install torch==2.5.1 torchaudio==2.5.1 --no-cache-dir --index-url https://download.pytorch.org/whl/cu124
+elif [ "$CUDA_VERSION" = "12.6" ]; then
+    "$HOME/miniconda3/bin/pip" install torch==2.6 torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/cu126
 fi
 
 "$HOME/miniconda3/bin/pip" cache purge