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: