version: '3.8'

services:
  ubuntu-ssh:
    build: 
      context: .
      dockerfile: Dockerfile.bgpt
    volumes:
      - ./src:/root/data
      - /mnt/sdb/zly/datas/Datas:/mnt/sdb/zly/datas
      - /data:/data
    container_name: ubuntu-ssh
    pull_policy: if_not_present
    tty: true
    restart: unless-stopped
    image: zly/cuda-bgpt:latest
    ports:
      - 3222:22
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
      - OLLAMA_ORIGINS="chrome-extension://*"
    networks:
      - network3
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

  # openai-whisper-asr-webservice:
  #   ports:
  #       - '9000:9000'
  #   volumes:
  #         - './whisper-large-v3:/data/whisper'
  #   environment:
  #       - ASR_MODEL=large
  #       - ASR_ENGINE=openai_whisper
  #       - ASR_MODEL_PATH=/data/whisper
  #   image: 'onerahmet/openai-whisper-asr-webservice:latest-gpu'
  #   deploy:
  #     resources:
  #       reservations:
  #         devices:
  #           - driver: nvidia
  #             count: 1
  #             capabilities: [gpu]


networks:
  network3:
    name: network3