version: '3.8'
services:
  ubuntumutation:
    build: 
      context: ..
      dockerfile: Dockerfile
      args: 
        CACHEBUST: 1
        CONDA_ENV_NAME: "mutation"
        PYTHON_VERSION: "3.11"
        TAG_VERSION: "22.04"
        HTTP_PROXY: "http://192.168.1.103:7897"
        HTTPS_PROXY: "http://192.168.1.103:7897"
    volumes:
      - ../:/work
    container_name: ubuntumutation
    pull_policy: if_not_present
    ulimits:
      memlock:
        soft: -1
        hard: -1
    restart: unless-stopped
    image: hotwa/mutation:testbuild
    privileged: true
    cap_add:
      - ALL
      - CAP_SYS_PTRACE
    shm_size: '64gb'
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
    command: ["/usr/sbin/sshd", "-D"]
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    networks:
      - network_ollama

networks:
  network_ollama:
    name: network_ollama