ROCM support in Docker

2025-11-18 09:52:18 -05:00 · 2024-04-10 16:25:11 +03:00
parent 015c2ec829
commit 8bf77f58f1
2 changed files with 83 additions and 0 deletions
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -33,6 +33,19 @@ services:
    environment:
      - CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api

+  auto-rocm:
+    <<: *base_service
+    profiles: ["auto-rocm"]
+    build: 
+      context: ./services/AUTOMATIC1111
+      dockerfile: Dockerfile.rocm
+    devices:
+      - "/dev/kfd"
+      - "/dev/dri"
+    deploy: {}
+    environment:
+      - CLI_ARGS=--allow-code --medvram --enable-insecure-extension-access --api
+
  auto-cpu:
    <<: *automatic
    profiles: ["auto-cpu"]