stable-diffusion-webui/docker-compose.amd.yml

services:
  stable-diffusion:
    build:
      args:
        PYTORCH_IMAGE: rocm/pytorch:rocm5.2.3_ubuntu20.04_py3.7_pytorch_1.12.1
    devices:
      - /dev/dri
      - /dev/kfd
Amd gpu docker (#1366) # Description AMD support can be built by prefixing your compose commands with `docker compose -f docker-compose.yml -f docker-compose.amd.yml ...` or, by setting ``` export COMPOSE_FILE=docker-compose.yml:docker-compose.amd.yml ``` in your `.profile` or through a tool like `direnv` Closes: Discord #installation-packing:AMD, at least for Linux hosts # Checklist: - [x] I have changed the base branch to `dev` - [x] I have performed a self-review of my own code - [x] I have commented my code in hard-to-understand areas - [x] I have made corresponding changes to the documentation 2022-10-01 01:42:35 +03:00			`services:`
			`stable-diffusion:`
			`build:`
			`args:`
			`PYTORCH_IMAGE: rocm/pytorch:rocm5.2.3_ubuntu20.04_py3.7_pytorch_1.12.1`
			`devices:`
			`- /dev/dri`
			`- /dev/kfd`