From 825c6899403e10bd4afbf0201f27aa074a4b6b31 Mon Sep 17 00:00:00 2001
From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com>
Date: Thu, 22 May 2025 17:27:43 +0300
Subject: [PATCH] Update docker_build.sh

---
 docker_build.sh | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/docker_build.sh b/docker_build.sh
index 4db652a1..354599d2 100644
--- a/docker_build.sh
+++ b/docker_build.sh
@@ -14,18 +14,18 @@ fi
 trap 'echo "Error Occured at \"$BASH_COMMAND\" with exit code $?"; exit 1' ERR
 
 LITE=false
-CUDA_VERSION=12.4
+CUDA_VERSION=12.6
 
 print_help() {
     echo "Usage: bash docker_build.sh [OPTIONS]"
     echo ""
     echo "Options:"
-    echo "  --cuda 12.4|12.8    Specify the CUDA VERSION (REQUIRED)"
+    echo "  --cuda 12.6|12.8    Specify the CUDA VERSION (REQUIRED)"
     echo "  --lite              Build a Lite Image"
     echo "  -h, --help          Show this help message and exit"
     echo ""
     echo "Examples:"
-    echo "  bash docker_build.sh --cuda 12.4 --funasr --faster-whisper"
+    echo "  bash docker_build.sh --cuda 12.6 --funasr --faster-whisper"
 }
 
 # Show help if no arguments provided
@@ -39,15 +39,15 @@ while [[ $# -gt 0 ]]; do
     case "$1" in
     --cuda)
         case "$2" in
-        12.4)
-            CUDA_VERSION=12.4
+        12.6)
+            CUDA_VERSION=12.6
             ;;
         12.8)
             CUDA_VERSION=12.8
             ;;
         *)
             echo "Error: Invalid CUDA_VERSION: $2"
-            echo "Choose From: [12.4, 12.8]"
+            echo "Choose From: [12.6, 12.8]"
             exit 1
             ;;
         esac