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
16 lines
356 B
Bash
16 lines
356 B
Bash
#!/usr/bin/env bash
|
|
|
|
project_name=${PROJECT_NAME:-invokeai}
|
|
volumename=${VOLUMENAME:-${project_name}_data}
|
|
arch=${ARCH:-x86_64}
|
|
platform=${PLATFORM:-Linux/${arch}}
|
|
invokeai_tag=${INVOKEAI_TAG:-${project_name}:${arch}}
|
|
gpus=${GPU_FLAGS:+--gpus=${GPU_FLAGS}}
|
|
|
|
export project_name
|
|
export volumename
|
|
export arch
|
|
export platform
|
|
export invokeai_tag
|
|
export gpus
|