江不江
		
	commited on
		
		
					Commit 
							
							·
						
						8fcd0f0
	
1
								Parent(s):
							
							b5e86a6
								
add docker-compose-gpu.yml to support gpu (#1591)
Browse files### 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 <[email protected]>
    	
        docker/docker-compose-gpu.yml
    ADDED
    
    | @@ -0,0 +1,37 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            include:
         | 
| 2 | 
            +
              - path: ./docker-compose-base.yml
         | 
| 3 | 
            +
                env_file: ./.env
         | 
| 4 | 
            +
             | 
| 5 | 
            +
            services:
         | 
| 6 | 
            +
              ragflow:
         | 
| 7 | 
            +
                depends_on:
         | 
| 8 | 
            +
                  mysql:
         | 
| 9 | 
            +
                    condition: service_healthy
         | 
| 10 | 
            +
                  es01:
         | 
| 11 | 
            +
                    condition: service_healthy
         | 
| 12 | 
            +
                image: infiniflow/ragflow:${RAGFLOW_VERSION}
         | 
| 13 | 
            +
                container_name: ragflow-server
         | 
| 14 | 
            +
                ports:
         | 
| 15 | 
            +
                  - ${SVR_HTTP_PORT}:9380
         | 
| 16 | 
            +
                  - 80:80
         | 
| 17 | 
            +
                  - 443:443
         | 
| 18 | 
            +
                volumes:
         | 
| 19 | 
            +
                  - ./service_conf.yaml:/ragflow/conf/service_conf.yaml
         | 
| 20 | 
            +
                  - ./ragflow-logs:/ragflow/logs
         | 
| 21 | 
            +
                  - ./nginx/ragflow.conf:/etc/nginx/conf.d/ragflow.conf
         | 
| 22 | 
            +
                  - ./nginx/proxy.conf:/etc/nginx/proxy.conf
         | 
| 23 | 
            +
                  - ./nginx/nginx.conf:/etc/nginx/nginx.conf
         | 
| 24 | 
            +
                environment:
         | 
| 25 | 
            +
                  - TZ=${TIMEZONE}
         | 
| 26 | 
            +
                  - HF_ENDPOINT=https://huggingface.co
         | 
| 27 | 
            +
                  - MACOS=${MACOS}
         | 
| 28 | 
            +
                networks:
         | 
| 29 | 
            +
                  - ragflow
         | 
| 30 | 
            +
                restart: always
         | 
| 31 | 
            +
                deploy:
         | 
| 32 | 
            +
                  resources:
         | 
| 33 | 
            +
                    reservations:
         | 
| 34 | 
            +
                      devices:
         | 
| 35 | 
            +
                        - driver: nvidia
         | 
| 36 | 
            +
                          count: all
         | 
| 37 | 
            +
                          capabilities: [gpu]
         |