diff --git a/docker/Dockerfile b/docker/Dockerfile index 2101820..4a53256 100644 --- 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/