From 50aa2e93717d5c768557b16ad711f9eba2bde2d2 Mon Sep 17 00:00:00 2001
From: Artsiom Siamashka <artsiom.siamashka@pivotics.com>
Date: Thu, 20 Nov 2025 11:59:31 +0100
Subject: [PATCH] Moved pytorch installation to position after other
 dependencies

---
 MLBaseDockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/MLBaseDockerfile b/MLBaseDockerfile
index 9f0fd1d..1751bb5 100644
--- a/MLBaseDockerfile
+++ b/MLBaseDockerfile
@@ -7,10 +7,10 @@ RUN pip install --root-user-action ignore huggingface_hub
 RUN hf download sentence-transformers/all-distilroberta-v1 --local-dir /assets/models/sentence-transformers/all-distilroberta-v1
 RUN hf download cross-encoder/qnli-distilroberta-base --local-dir /assets/models/cross-encoder/qnli-distilroberta-base
 
-RUN pip install --root-user-action ignore torch torchvision --index-url https://download.pytorch.org/whl/cu130
 RUN conda install -c conda-forge \
     numpy=1.26.4 \
     pandas=2.3.3 \
     sentence-transformers=3.0.1 \
     keybert=0.9.0 \
     faiss-gpu=1.9.0
+RUN pip install --root-user-action ignore torch torchvision --index-url https://download.pytorch.org/whl/cu130