first pass

templates/python/ml-pytorch/docker-compose.yml.template

@@ -0,0 +1,28 @@
+services:
+  {{PROJECT_NAME}}:
+    build: .
+    container_name: {{PROJECT_NAME}}
+    restart: unless-stopped
+    ports:
+      - "${HOST_PORT:-{{PORT}}}:{{PORT}}"
+{{#if USE_ENV_FILE}}
+    env_file:
+      - .env
+{{/if}}
+    environment:
+      PYTHONUNBUFFERED: 1
+      PORT: {{PORT}}
+{{#if HAS_VOLUMES}}
+    volumes:
+{{#each VOLUMES}}
+      - {{this}}
+{{/each}}
+{{/if}}
+    # Uncomment below for GPU support
+    # deploy:
+    #   resources:
+    #     reservations:
+    #       devices:
+    #         - driver: nvidia
+    #           count: 1
+    #           capabilities: [gpu]