FIX FAISS

2024-10-04 19:48:54 +02:00
1 changed files with 1 additions and 1 deletions
--- a/docker/Dockerfile
+++ b/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/