From 7f80a928e2917f757511e18ea92cea87388325a3 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Thu, 30 May 2024 19:35:56 -0700 Subject: [PATCH] Update Dockerfile --- Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index e35d695a..c7a23c97 100644 --- a/Dockerfile +++ b/Dockerfile @@ -14,7 +14,7 @@ ENV ENV=prod \ # Install GCC and build tools RUN apt-get update && \ - apt-get install -y gcc build-essential pkg-config && \ + apt-get install -y gcc build-essential pkg-config libhdf5-dev && \ apt-get clean && \ rm -rf /var/lib/apt/lists/* @@ -27,6 +27,7 @@ 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 && \ + pip3 install tensorflow[and-cuda] --no-cache-dir && \ uv pip install --system -r requirements.txt --no-cache-dir; \ else \ pip3 install --no-binary=h5py h5py --no-cache-dir && \