mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-04-23 03:00:31 -04:00
eb702a5049
env.sh: - move check for torch to CONVTAINER_FLAVOR detection Dockerfile - only mount `/var/cache/apt` for apt related steps - remove `docker-clean` from `/etc/apt/apt.conf.d` for BuildKit cache - remove apt-get clean for BuildKit cache - only copy frontend to frontend-builder - mount `/usr/local/share/.cache/yarn` in frountend-builder - separate steps for yarn install and yarn build - build pytorch in pyproject-builder build.sh - prepare for installation with extras
45 lines
1.4 KiB
Bash
Executable File
45 lines
1.4 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -e
|
|
|
|
# How to use: https://invoke-ai.github.io/InvokeAI/installation/INSTALL_DOCKER/#setup
|
|
# Some possible pip extra-index urls (cuda 11.7 is available without extra url):
|
|
# CUDA 11.6: https://download.pytorch.org/whl/cu116
|
|
# ROCm 5.2: https://download.pytorch.org/whl/rocm5.2
|
|
# CPU: https://download.pytorch.org/whl/cpu
|
|
# as found on https://pytorch.org/get-started/locally/
|
|
|
|
SCRIPTDIR=$(dirname "$0")
|
|
cd "$SCRIPTDIR" || exit 1
|
|
|
|
source ./env.sh
|
|
|
|
DOCKERFILE=${INVOKE_DOCKERFILE:-Dockerfile}
|
|
|
|
# print the settings
|
|
echo -e "You are using these values:\n"
|
|
echo -e "Dockerfile: \t${DOCKERFILE}"
|
|
echo -e "index-url: \t${PIP_EXTRA_INDEX_URL:-none}"
|
|
echo -e "Volumename: \t${VOLUMENAME}"
|
|
echo -e "Platform: \t${PLATFORM}"
|
|
echo -e "Registry: \t${CONTAINER_REGISTRY}"
|
|
echo -e "Repository: \t${CONTAINER_REPOSITORY}"
|
|
echo -e "Container Tag: \t${CONTAINER_TAG}"
|
|
echo -e "Container Image: ${CONTAINER_IMAGE}\n"
|
|
|
|
# Create docker volume
|
|
if [[ -n "$(docker volume ls -f name="${VOLUMENAME}" -q)" ]]; then
|
|
echo -e "Volume already exists\n"
|
|
else
|
|
echo -n "createing docker volume "
|
|
docker volume create "${VOLUMENAME}"
|
|
fi
|
|
|
|
# Build Container
|
|
DOCKER_BUILDKIT=1 docker build \
|
|
--platform="${PLATFORM}" \
|
|
--tag="${CONTAINER_IMAGE}" \
|
|
${PIP_EXTRA_INDEX_URL:+--build-arg="PIP_EXTRA_INDEX_URL=${PIP_EXTRA_INDEX_URL}"} \
|
|
${PIP_PACKAGE:+--build-arg="PIP_PACKAGE=${PIP_PACKAGE}"} \
|
|
--file="${DOCKERFILE}" \
|
|
..
|