Readme.md hinzugefügt

.gitea/workflows/docker-build.yml

@@ -14,8 +14,17 @@ jobs:
 
     - name: Get repository code
       uses: actions/checkout@v4
+
+    - name: ARM64 add QEMU 
+      uses: docker/setup-qemu-action@v3
+      with:
+        platforms: arm64
+    - name: ARM64 setup Docker buildx 
+      uses: docker/setup-buildx-action@v3
+
     - name: Build the images
       run: bash ./build-images.sh -no-updates
+
     - name: Docker login
       uses: docker/login-action@v3
       with:

Readme.md

@@ -0,0 +1,5 @@
+> [!NOTE]  
+> In diesem Repository befinden sich die Quelldaten und Skripte zur Erstellung der Docker-Base-Images.
+
+> [!WARNING]
+> Um die Plattform zu benutzen, bitte das [Agent-Template](https://git.chai.uni-hamburg.de/UMS-Agenten/Agent-Template) benutzen!

docker/Dockerfile

@@ -37,13 +37,13 @@ RUN if [ ! -d /usr/local/cuda/ ]; then \
 			faiss-cpu; \
 	else \
 		python3 -m pip install --no-cache-dir --break-system-packages \
-			faiss-gpu; \
+			faiss-gpu-cu12; \
 	fi;
 
-RUN mkdir /ums-agenten/
+RUN mkdir -p /ums-agenten/persist/
 
-ENV NLTK_DATA=/ums-agenten/nltk/
+ENV NLTK_DATA=/ums-agenten/persist/nltk/
-ENV HF_HOME=/ums-agenten/hf_home/
+ENV HF_HOME=/ums-agenten/persist/hf_home/
 
 COPY ./docker/$PIP_REQ_FILE /ums-agenten/requirements.txt
 RUN pip3 install --break-system-packages --no-cache-dir -r /ums-agenten/requirements.txt \

docker/Dockerfile.gpu

@@ -23,10 +23,6 @@ RUN apt update && \
 RUN python3 -m pip install --no-cache-dir --break-system-packages \
     torch --index-url https://download.pytorch.org/whl/cu124
 
-RUN git clone https://github.com/NVIDIA/apex
-RUN cd apex && \
-    pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --break-system-packages --config-settings  "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./
-
 # fix library path for tensorflow and link a file
 ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/usr/local/cuda/lib64"
 RUN ln -s /usr/local/cuda/lib64/libcusolver.so.11 /usr/local/cuda/lib64/libcusolver.so.10