Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

(docker) dockerfile fixes including PR feedback

Browse Source 
When previously using base Debian-ish images, the Invoke image
failed to find CUDA drivers on some RHEL-ish distributions
This commit is contained in:
Eugene Brodsky 2023-07-11 20:57:33 -04:00
parent 7111db2e0d
commit d6da7ad922
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
docker/Dockerfile
View File

@ -2,7 +2,7 @@
## Builder stage ## Builder stage
FROM ubuntu:22.04 AS builder FROM library/ubuntu:22.04 AS builder
ARG DEBIAN_FRONTEND=noninteractive ARG DEBIAN_FRONTEND=noninteractive
RUN rm -f /etc/apt/apt.conf.d/docker-clean; echo 'Binary::apt::APT::Keep-Downloaded-Packages "true";' > /etc/apt/apt.conf.d/keep-cache RUN rm -f /etc/apt/apt.conf.d/docker-clean; echo 'Binary::apt::APT::Keep-Downloaded-Packages "true";' > /etc/apt/apt.conf.d/keep-cache
@ -37,7 +37,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
elif [ "$GPU_DRIVER" = "rocm" ]; then \ elif [ "$GPU_DRIVER" = "rocm" ]; then \
extra_index_url_arg="--extra-index-url https://download.pytorch.org/whl/rocm5.2"; \ extra_index_url_arg="--extra-index-url https://download.pytorch.org/whl/rocm5.2"; \
else \ else \
extra_index_url_arg="--extra-index-url https://download.pytorch.org/whl/cu117"; \ extra_index_url_arg="--extra-index-url https://download.pytorch.org/whl/cu118"; \
fi &&\ fi &&\
pip install $extra_index_url_arg \ pip install $extra_index_url_arg \
torch==$TORCH_VERSION \ torch==$TORCH_VERSION \
@ -59,18 +59,18 @@ RUN --mount=type=cache,target=/root/.cache/pip \
# #### Build the Web UI ------------------------------------ # #### Build the Web UI ------------------------------------
FROM node:18 as web-builder FROM node:18 AS web-builder
WORKDIR /build WORKDIR /build
COPY invokeai/frontend/web/ ./ COPY invokeai/frontend/web/ ./
RUN --mount=type=cache,target=node_modules \ RUN --mount=type=cache,target=/usr/lib/node_modules \
npm install --include dev npm install --include dev
RUN --mount=type=cache,target=node_modules \ RUN --mount=type=cache,target=/usr/lib/node_modules \
yarn vite build yarn vite build
#### Runtime stage --------------------------------------- #### Runtime stage ---------------------------------------
FROM library/ubuntu:22.04 as runtime FROM library/ubuntu:22.04 AS runtime
ARG DEBIAN_FRONTEND=noninteractive ARG DEBIAN_FRONTEND=noninteractive
ENV PYTHONUNBUFFERED=1 ENV PYTHONUNBUFFERED=1