diff --git a/MLBaseDockerfile b/MLBaseDockerfile
index fbd9f63..9f0fd1d 100644
--- a/MLBaseDockerfile
+++ b/MLBaseDockerfile
@@ -7,6 +7,7 @@ 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 \