feat: don't use gpu when execute cpu mode (#43)

1 year ago · 04c3ec5d3b
parent 979b9d611e
commit 04c3ec5d3b
1 changed files with 6 additions and 2 deletions
--- a/docker-compose.yml
+++ b/docker-compose.yml
@ -14,10 +14,11 @@ services:
      - "4500:4500"
      - "6500:6500"
      - "8000:8000" # eval port
-    env_file:
-      - .env
    environment:
+      - USE_GPU=False
      - CELERY_BROKER_URL=redis://redis:6379
+    env_file:
+      - .env
    depends_on:
      - redis

@ -35,6 +36,9 @@ services:
      - "4500:4500"
      - "6500:6500"
      - "8000:8000" # eval port
+    environment:
+      - USE_GPU=True
+      - CELERY_BROKER_URL=redis://redis:6379
    env_file:
      - .env
    deploy: