Update Dockerfile

open-webui · May 31, 2024 · f982d66 · f982d66
1 parent d48e51c
commit f982d66
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -25,9 +25,11 @@ WORKDIR /app
 # Install Python dependencies
 RUN pip3 install uv && \
     if [ "$USE_CUDA" = "true" ]; then \
+    pip3 install --no-binary=h5py h5py --no-cache-dir && \
     pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/$USE_CUDA_DOCKER_VER --no-cache-dir && \
     uv pip install --system -r requirements.txt --no-cache-dir; \
     else \
+    pip3 install --no-binary=h5py h5py --no-cache-dir && \
     pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu --no-cache-dir && \
     uv pip install --system -r requirements.txt --no-cache-dir; \
     fi