diff --git a/docker/docker-compose-gpu.yml b/docker/docker-compose-gpu.yml new file mode 100644 index 000000000..8681bad80 --- /dev/null +++ b/docker/docker-compose-gpu.yml @@ -0,0 +1,37 @@ +include: + - path: ./docker-compose-base.yml + env_file: ./.env + +services: + ragflow: + depends_on: + mysql: + condition: service_healthy + es01: + condition: service_healthy + image: infiniflow/ragflow:${RAGFLOW_VERSION} + container_name: ragflow-server + ports: + - ${SVR_HTTP_PORT}:9380 + - 80:80 + - 443:443 + volumes: + - ./service_conf.yaml:/ragflow/conf/service_conf.yaml + - ./ragflow-logs:/ragflow/logs + - ./nginx/ragflow.conf:/etc/nginx/conf.d/ragflow.conf + - ./nginx/proxy.conf:/etc/nginx/proxy.conf + - ./nginx/nginx.conf:/etc/nginx/nginx.conf + environment: + - TZ=${TIMEZONE} + - HF_ENDPOINT=https://huggingface.co + - MACOS=${MACOS} + networks: + - ragflow + restart: always + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: all + capabilities: [gpu]