version: '3' # slightly different to make sure CPU tests run without nvidia device
services:
  gpt-neox:
    command: nvidia-smi dmon
    image: gpt-neox
    build:
      context: .
      dockerfile: Dockerfile
    shm_size: 1g
    ulimits:
      memlock:
        soft: -1
        hard: -1
    logging:
      options:
        max-size: "100m"
        max-file: "3"
    volumes:
      - ${NEOX_DATA_PATH}:/home/mchorse/data
      - ${NEOX_CHECKPOINT_PATH}:/home/mchorse/chk
      - .:/home/mchorse/gpt-neox