UMS-Agenten/Base-Images
3
0
Fork 0
Code Actions Packages Releases Activity

Fixes for Actions
All checks were successful
Build and push Docker image in git push / build (push) Successful in 4m58s
Details

Browse Source
This commit is contained in:
Magnus Bender
2024-10-04 17:47:12 +02:00
parent fa36f93466
commit fdbbbc3e3d
5 changed files with 16 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
docker/Dockerfile
View File

@ -29,8 +29,16 @@ RUN apt-get update \
curl wget \
htop screen
RUN python3 -m pip install --no-cache-dir --break-system-packages \
torch --index-url https://download.pytorch.org/whl/cpu
# if cpu container (install cpu only versions; for gpu, gpu versions)
RUN if [ ! -d /usr/local/cuda/ ]; then \
python3 -m pip install --no-cache-dir --break-system-packages \
torch --index-url https://download.pytorch.org/whl/cpu; \
python3 -m pip install --no-cache-dir --break-system-packages \
faiss-cpu; \
else \
python3 -m pip install --no-cache-dir --break-system-packages \
faiss-gpu; \
fi;
RUN mkdir /ums-agenten/