.gitea/workflows/docker-build.yml

@@ -14,17 +14,8 @@ 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

@@ -1,5 +0,0 @@
-> [!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-cu12; \
+			faiss-gpu; \
 	fi;
 
-RUN mkdir -p /ums-agenten/persist/
+RUN mkdir /ums-agenten/
 
-ENV NLTK_DATA=/ums-agenten/persist/nltk/
+ENV NLTK_DATA=/ums-agenten/nltk/
-ENV HF_HOME=/ums-agenten/persist/hf_home/
+ENV HF_HOME=/ums-agenten/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,6 +23,10 @@ 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