Fixes for Actions

docker/Dockerfile

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

docker/Dockerfile.gpu

@@ -20,12 +20,12 @@ RUN apt update && \
 		libopenmpi-dev && \
 	rm -rf /var/lib/apt/lists
 
-RUN python3 -m pip install --no-cache-dir \
+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 --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./
+    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"

docker/requirements.txt

@@ -17,7 +17,6 @@ matplotlib
 
 transformers
 huggingface_hub
-faiss-cpu
 datasets
 accelerate
 peft