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

Moved pytorch installation to position after other dependencies

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

2
MLBaseDockerfile
View File

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