From a4d230f12bc76fbcfc9b3948cfa2dda54d3bb40e Mon Sep 17 00:00:00 2001 From: zhuhao <37029601+hwzhuhao@users.noreply.github.com> Date: Mon, 22 Jul 2024 09:25:42 +0800 Subject: [PATCH] add docker-compose-gpu-CN.yml and docker-compose-gpu-CN-oc9.yml to support gpu (#1618) ### What problem does this PR solve? add docker-compose-gpu-CN.yml and docker-compose-gpu-CN-oc9.yml to support gpu #1558 ### Type of change - [x] New Feature (non-breaking change which adds functionality) --- docker/docker-compose-gpu-CN-oc9.yml | 37 ++++++++++++++++++++++++++++ docker/docker-compose-gpu-CN.yml | 37 ++++++++++++++++++++++++++++ 2 files changed, 74 insertions(+) create mode 100644 docker/docker-compose-gpu-CN-oc9.yml create mode 100644 docker/docker-compose-gpu-CN.yml diff --git a/docker/docker-compose-gpu-CN-oc9.yml b/docker/docker-compose-gpu-CN-oc9.yml new file mode 100644 index 000000000..1950c62cd --- /dev/null +++ b/docker/docker-compose-gpu-CN-oc9.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: edwardelric233/ragflow:oc9 + 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://hf-mirror.com + - MACOS=${MACOS} + networks: + - ragflow + restart: always + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: all + capabilities: [gpu] diff --git a/docker/docker-compose-gpu-CN.yml b/docker/docker-compose-gpu-CN.yml new file mode 100644 index 000000000..e061a6897 --- /dev/null +++ b/docker/docker-compose-gpu-CN.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: swr.cn-north-4.myhuaweicloud.com/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://hf-mirror.com + - MACOS=${MACOS} + networks: + - ragflow + restart: always + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: all + capabilities: [gpu]