version: '3.8'

services:
  ubuntu-cpp:
    build: 
      context: .
      dockerfile: Dockerfile.bgpt
    volumes:
      - /data:/data/
      - /home/chipan:/home/chipan
    container_name: ubuntu-cpp
    pull_policy: if_not_present
    tty: true
    restart: unless-stopped
    image: zly/cuda-bgpt:latest
    ports:
      - 3111:22
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
      - OLLAMA_ORIGINS="chrome-extension://*"
    networks:
      - network4
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

networks:
  network4:
    name: network4