intent/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
      - /root/aibot/intent/m3e:/home/model-server/m3e
    ports:
      - "50072:50072"
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              #count: 1
              capabilities: [gpu]
    ipc: host
    command: ["/bin/sh", "-c", "python web.py"]