version: '3.8'

services:
  gromacs:
    build: 
      context: .
      dockerfile: Dockerfile.3.gromacs
      args: 
        CACHEBUST: 1
        BASE: "hotwa/gromacs"
        TAG_VERSION: "micromamba"
        PLUMED_VERSION: "2.9.2"
        FFTW_VERSION: "3.3.10"
        GROMACS_VERSION: "2021.7"
        CUDA_ARCH: "75;86;89"
        BUILDKIT_INLINE_CACHE: 1
    # env_file:
    #   - .env
    container_name: cuda_3
    pull_policy: if_not_present
    ulimits:
      memlock:
        soft: -1
        hard: -1
    restart: unless-stopped
    image: hotwa/gromacs:gromacs
    # privileged: true
    # cap_add:
    #   - ALL
    #   - CAP_SYS_PTRACE
    # shm_size: '16gb'
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
      - TMPDIR=/var/tmp
    # ports:
    #   - "53322:2222" 
    # network_mode: host
    # command: ["/usr/sbin/sshd", "-D"]
    command: ["tail", "-f", "/dev/null"]
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]