first add

docker/docker-compose.yml

@@ -0,0 +1,16 @@
+version: '3.8'
+
+services:
+  ml-training:
+    build:
+      context: ..
+      dockerfile: docker/Dockerfile
+    volumes:
+      - ../:/app
+    environment:
+      - PYTHONUNBUFFERED=1
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]