lxbfYeaa/AIServicesBaseImages
Archived
Template
1
0
Fork 0
forked from LiveCarta/AIServicesBaseImages
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added command to install the latest pytorch for CUDA 13

Browse Source
This commit is contained in:
Artsiom Siamashka
2025-11-20 11:56:30 +01:00
parent f6a5fbdfe5
commit 5d94c77d3b
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
MLBaseDockerfile
View File

@@ -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 \