intent/docker/docker-compose.yaml

version: '3.5'
services:
  intent:
    container_name: ai-slibra-intent
    image: tiangolo/gunicorn:latest
    entrypoint: python web.py
    restart: always
    user: 0:0
    environment:
      - CUDA_VISIBLE_DEVICES=0
      - TZ=Asia/Shanghai  # 设置时区
    volumes:
      - /root/aibot/intent/models:/home/model-server/intent
      - /root/aibot/intent/m3e:/home/model-server/m3e
    ports:
      - "50072:50072"
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              #count: 1
              capabilities: [gpu]