first cimmit

docker-compose.yml

@@ -0,0 +1,25 @@
+services:
+  AiDA_Model_Litserve:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    working_dir: /app
+    environment:
+      GOOGLE_APPLICATION_CREDENTIALS: /google_application_credentials.json
+      DEBUG: 0
+    volumes:
+      - ./app:/app/app
+      - ./.prod_env:/app/.env
+      - ./data:/data
+      - ./google_application_credentials.json:/google_application_credentials.json
+      - /etc/localtime:/etc/localtime:ro
+    ports:
+      - "10070:8000"
+    deploy:
+      resources:
+        reservations:
+          devices:
+            # 告诉 Docker 使用所有可用的 NVIDIA GPU
+            - driver: nvidia
+              device_ids: [ '0' ]
+              capabilities: [ gpu ]