mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-11-11 03:24:30 +00:00
ci: add ROCm Dockerfile (#2094)
This commit is contained in:
1 parent
d637cfb935
commit
7e3e20a50b
2 files changed
+82
-1
No files matched your search
+32
-1
@@ -24,12 +24,25 @@ docker build -f docker/Dockerfile -t sd .
|
||||
|
||||
## Building variants using Docker
|
||||
|
||||
Vulkan:
|
||||
### Vulkan:
|
||||
|
||||
```shell
|
||||
docker build -f docker/Dockerfile.vulkan -t sd .
|
||||
```
|
||||
|
||||
### ROCm (AMD GPU):
|
||||
|
||||
Defining one or multiple targets with `--build-arg AMDGPU_TARGETS` is strongly recommended.
|
||||
Building the image will otherwise likely fail or build for a target that does not fit your
|
||||
hardware.
|
||||
|
||||
```shell
|
||||
# Build for one target
|
||||
docker build -f docker/Dockerfile.rocm --build-arg "AMDGPU_TARGETS=gfx1100" -t sd .
|
||||
# Build for multiple targets
|
||||
docker build -f docker/Dockerfile.rocm --build-arg "AMDGPU_TARGETS=gfx1030;gfx1100" -t sd .
|
||||
```
|
||||
|
||||
## Run locally built image's CLI
|
||||
|
||||
```shell
|
||||
@@ -37,3 +50,21 @@ docker run --rm -v /path/to/models:/models -v /path/to/output/:/output sd [args.
|
||||
# For example
|
||||
# docker run --rm -v ./models:/models -v ./build:/output sd -m /models/sd-v1-4.ckpt -p "a lovely cat" -v -o /output/output.png
|
||||
```
|
||||
|
||||
### ROCm (AMD GPU)
|
||||
|
||||
The user that executes this command needs to be in the `video` group and have read and write
|
||||
access to `/dev/dri` and `/dev/kfd`.
|
||||
|
||||
`--device` and `--group-add` allow the container access to the GPU.
|
||||
|
||||
```shell
|
||||
docker run --rm --device=/dev/dri --device=/dev/kfd --group-add video \
|
||||
-v /path/to/models:/models -v /path/to/output/:/output \
|
||||
sd [args...]
|
||||
# For example
|
||||
# docker run --rm --device=/dev/dri --device=/dev/kfd --group-add video \
|
||||
# -v ./models:/models -v ./build:/output \
|
||||
# sd -m /models/sd-v1-4.ckpt -v -o /output/output.png \
|
||||
# -p "a lovely cat"
|
||||
```
|
||||
Reference in new issue
Block a user