From d33dde7fc8c27ca0e78ff98efff0f71e23d88632 Mon Sep 17 00:00:00 2001
From: ekoc <erol.koc@ergon.ch>
Date: Mon, 9 Jun 2025 13:41:55 +0200
Subject: [PATCH] Fix pytorch version to 2.6.0

---
 docker/base/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/base/Dockerfile b/docker/base/Dockerfile
index 91d119b..59a2a51 100644
--- a/docker/base/Dockerfile
+++ b/docker/base/Dockerfile
@@ -76,7 +76,7 @@ RUN <<EOR
   conda activate alltalk
   mkdir -p ${ALLTALK_DIR}/pip_cache
   pip install --no-cache-dir --cache-dir=${ALLTALK_DIR}/pip_cache \
-      torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu${CUDA_SHORT_VERSION_NO_DOT}
+      torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu${CUDA_SHORT_VERSION_NO_DOT}
 
   if [ $? -ne 0 ] ; then
     echo "Failed to install pip dependencies: $RESULT"