OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-06-26 18:15:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update Dockerfile

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-30 19:35:56 -07:00
parent f982d667d7
commit 7f80a928e2
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Dockerfile
View File

@ -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 && \