Support for ARM64

.gitea/workflows/docker-build.yml

@@ -14,8 +14,17 @@ jobs:
 
     - name: Get repository code
       uses: actions/checkout@v4
+
+    - name: ARM64 support: QEMU 
+      uses: docker/setup-qemu-action@v3
+      with:
+        platforms: arm64
+    - name: ARM64 support: 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:

docker/Dockerfile

@@ -37,7 +37,7 @@ 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/

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