From 0b94376cd4df8f7ec188479276c7201e640781f1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B1=9F=E4=B8=8D=E6=B1=9F?=
 <74400272+Seaver-Zhu@users.noreply.github.com>
Date: Thu, 18 Jul 2024 15:45:12 +0800
Subject: [PATCH] add docker-compose-gpu.yml to support gpu (#1591)

### What problem does this PR solve?

add docker-compose-gpu.yml to support gpu
#1558

### Type of change
- [x] New Feature (non-breaking change which adds functionality)

---------

Signed-off-by: seaver <zhudan187@qq.com>
---
 docker/docker-compose-gpu.yml | 37 +++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 docker/docker-compose-gpu.yml

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]