File size: 506 Bytes
d90b3a8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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
|