mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
c5aeb36230
With these changes, the Docker image can be built and executed successfully on hosts with AMD devices with ROCm acceleration. Previously, a ROCm-enabled version of torch would be installed, but later removed during installation of InvokeAI itself. This was caused by InvokeAI needing a newer torch version than was previously installed. The fix consists of multiple components: * Update the hardcoded versions of torch and torchvision to the versions currently used in pyproject.toml, so that a new version need not be installed during installation of InvokeAI. * Specify --extra-index-url on installation of InvokeAI so that even if a verison mismatch occurs, the correct torch version should still be installed. This also necessitates changing --index-url to --extra-index-url for the Torch repo. Otherwise non-torch dependencies would not be found. * In run.sh, build the image for the selected service.
129 lines
4.0 KiB
Docker
129 lines
4.0 KiB
Docker
# syntax=docker/dockerfile:1.4
|
|
|
|
## Builder stage
|
|
|
|
FROM library/ubuntu:23.04 AS builder
|
|
|
|
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 --mount=type=cache,target=/var/cache/apt,sharing=locked \
|
|
--mount=type=cache,target=/var/lib/apt,sharing=locked \
|
|
apt update && apt-get install -y \
|
|
git \
|
|
python3-venv \
|
|
python3-pip \
|
|
build-essential
|
|
|
|
ENV INVOKEAI_SRC=/opt/invokeai
|
|
ENV VIRTUAL_ENV=/opt/venv/invokeai
|
|
|
|
ENV PATH="$VIRTUAL_ENV/bin:$PATH"
|
|
ARG TORCH_VERSION=2.1.2
|
|
ARG TORCHVISION_VERSION=0.16.2
|
|
ARG GPU_DRIVER=cuda
|
|
ARG TARGETPLATFORM="linux/amd64"
|
|
# unused but available
|
|
ARG BUILDPLATFORM
|
|
|
|
WORKDIR ${INVOKEAI_SRC}
|
|
|
|
# Install pytorch before all other pip packages
|
|
# NOTE: there are no pytorch builds for arm64 + cuda, only cpu
|
|
# x86_64/CUDA is default
|
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
python3 -m venv ${VIRTUAL_ENV} &&\
|
|
if [ "$TARGETPLATFORM" = "linux/arm64" ] || [ "$GPU_DRIVER" = "cpu" ]; then \
|
|
extra_index_url_arg="--extra-index-url https://download.pytorch.org/whl/cpu"; \
|
|
elif [ "$GPU_DRIVER" = "rocm" ]; then \
|
|
extra_index_url_arg="--extra-index-url https://download.pytorch.org/whl/rocm5.6"; \
|
|
else \
|
|
extra_index_url_arg="--extra-index-url https://download.pytorch.org/whl/cu121"; \
|
|
fi &&\
|
|
pip install $extra_index_url_arg \
|
|
torch==$TORCH_VERSION \
|
|
torchvision==$TORCHVISION_VERSION
|
|
|
|
# Install the local package.
|
|
# Editable mode helps use the same image for development:
|
|
# the local working copy can be bind-mounted into the image
|
|
# at path defined by ${INVOKEAI_SRC}
|
|
COPY invokeai ./invokeai
|
|
COPY pyproject.toml ./
|
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
# xformers + triton fails to install on arm64
|
|
if [ "$GPU_DRIVER" = "cuda" ] && [ "$TARGETPLATFORM" = "linux/amd64" ]; then \
|
|
pip install -e ".[xformers]"; \
|
|
else \
|
|
pip install $extra_index_url_arg -e "."; \
|
|
fi
|
|
|
|
# #### Build the Web UI ------------------------------------
|
|
|
|
FROM node:20-slim AS web-builder
|
|
ENV PNPM_HOME="/pnpm"
|
|
ENV PATH="$PNPM_HOME:$PATH"
|
|
RUN corepack enable
|
|
|
|
WORKDIR /build
|
|
COPY invokeai/frontend/web/ ./
|
|
RUN --mount=type=cache,target=/pnpm/store \
|
|
pnpm install --frozen-lockfile
|
|
RUN npx vite build
|
|
|
|
#### Runtime stage ---------------------------------------
|
|
|
|
FROM library/ubuntu:23.04 AS runtime
|
|
|
|
ARG DEBIAN_FRONTEND=noninteractive
|
|
ENV PYTHONUNBUFFERED=1
|
|
ENV PYTHONDONTWRITEBYTECODE=1
|
|
|
|
RUN apt update && apt install -y --no-install-recommends \
|
|
git \
|
|
curl \
|
|
vim \
|
|
tmux \
|
|
ncdu \
|
|
iotop \
|
|
bzip2 \
|
|
gosu \
|
|
magic-wormhole \
|
|
libglib2.0-0 \
|
|
libgl1-mesa-glx \
|
|
python3-venv \
|
|
python3-pip \
|
|
build-essential \
|
|
libopencv-dev \
|
|
libstdc++-10-dev &&\
|
|
apt-get clean && apt-get autoclean
|
|
|
|
|
|
ENV INVOKEAI_SRC=/opt/invokeai
|
|
ENV VIRTUAL_ENV=/opt/venv/invokeai
|
|
ENV INVOKEAI_ROOT=/invokeai
|
|
ENV PATH="$VIRTUAL_ENV/bin:$INVOKEAI_SRC:$PATH"
|
|
ENV CONTAINER_UID=${CONTAINER_UID:-1000}
|
|
ENV CONTAINER_GID=${CONTAINER_GID:-1000}
|
|
|
|
# --link requires buldkit w/ dockerfile syntax 1.4
|
|
COPY --link --from=builder ${INVOKEAI_SRC} ${INVOKEAI_SRC}
|
|
COPY --link --from=builder ${VIRTUAL_ENV} ${VIRTUAL_ENV}
|
|
COPY --link --from=web-builder /build/dist ${INVOKEAI_SRC}/invokeai/frontend/web/dist
|
|
|
|
# Link amdgpu.ids for ROCm builds
|
|
# contributed by https://github.com/Rubonnek
|
|
RUN mkdir -p "/opt/amdgpu/share/libdrm" &&\
|
|
ln -s "/usr/share/libdrm/amdgpu.ids" "/opt/amdgpu/share/libdrm/amdgpu.ids"
|
|
|
|
WORKDIR ${INVOKEAI_SRC}
|
|
|
|
# build patchmatch
|
|
RUN cd /usr/lib/$(uname -p)-linux-gnu/pkgconfig/ && ln -sf opencv4.pc opencv.pc
|
|
RUN python3 -c "from patchmatch import patch_match"
|
|
|
|
RUN mkdir -p ${INVOKEAI_ROOT} && chown -R ${CONTAINER_UID}:${CONTAINER_GID} ${INVOKEAI_ROOT}
|
|
|
|
COPY docker/docker-entrypoint.sh ./
|
|
ENTRYPOINT ["/opt/invokeai/docker-entrypoint.sh"]
|
|
CMD ["invokeai-web", "--host", "0.0.0.0"]
|