From 5d94c77d3b8c598d499dceb8b08757a8a25379bc Mon Sep 17 00:00:00 2001 From: Artsiom Siamashka Date: Thu, 20 Nov 2025 11:56:30 +0100 Subject: [PATCH] Added command to install the latest pytorch for CUDA 13 --- MLBaseDockerfile | 1 + 1 file changed, 1 insertion(+) 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 \