Add amd gpu support

Testing

docker-compose.yml

@@ -25,6 +25,13 @@ x-nvidia_docker: &nvidia_docker
             device_ids: ["0"]
             capabilities: [gpu]
 
+x-base_service_amd: &amd_docker
+  group_add:
+    - video
+  devices:
+    - "/dev/dri"
+    - "/dev/kfd"
+
 
 services:
   nvidia:
@@ -32,6 +39,11 @@ services:
     <<: *nvidia_docker
     profiles: ["nvidia"]
 
+  amd:
+    <<: *base_service
+    <<: *amd_docker
+    profiles: ["amd"]
+
   cpu:
     <<: *base_service
     profiles: ["cpu"]