From f982d667d7fe3fd8c1475e8ceac2b93e1a1f874f Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Thu, 30 May 2024 19:26:40 -0700
Subject: [PATCH] Update Dockerfile

---
 Dockerfile | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 47b96cbf..e35d695a 100644
--- 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