File size: 545 Bytes
d90b3a8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
version: '3'
services:
gpt-neox:
command: nvidia-smi dmon
image: leogao2/gpt-neox:main
shm_size: 1g
ulimits:
memlock:
soft: -1
hard: -1
runtime: nvidia
deploy:
resources:
reservations:
devices:
- driver: nvidia
capabilities: [gpu]
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
|