mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
5d20f47993
* Add gpu support to docker Enable GPUs within docker * Use gpus flag * Add GPU information to readme * Fix env var name for GPU
22 lines
498 B
Bash
Executable File
22 lines
498 B
Bash
Executable File
#!/usr/bin/env bash
|
|
set -e
|
|
|
|
source ./docker-build/env.sh || echo "please run from repository root" || exit 1
|
|
|
|
echo -e "You are using these values:\n"
|
|
echo -e "volumename:\t ${volumename}"
|
|
echo -e "invokeai_tag:\t ${invokeai_tag}\n"
|
|
|
|
docker run \
|
|
--interactive \
|
|
--tty \
|
|
--rm \
|
|
--platform="$platform" \
|
|
--name="$project_name" \
|
|
--hostname="$project_name" \
|
|
--mount="source=$volumename,target=/data" \
|
|
--publish=9090:9090 \
|
|
--cap-add=sys_nice \
|
|
$gpus \
|
|
"$invokeai_tag" ${1:+$@}
|