Skip to content

Instantly share code, notes, and snippets.

@butsugiri
Created August 8, 2022 08:46
Show Gist options
  • Save butsugiri/9c097c320ac61f5a6ef0483bf777adc1 to your computer and use it in GitHub Desktop.
Save butsugiri/9c097c320ac61f5a6ef0483bf777adc1 to your computer and use it in GitHub Desktop.
services:
fairseq:
build: .
volumes:
- type: bind
source: "/work00"
target: "/work00"
- type: bind
source: "./fairseq"
target: "/code/fairseq"
- type: bind
source: "./shell"
target: "/code/shell"
tty: true
shm_size: '2gb'
environment:
HOST_USERNAME: ${HOST_USERNAME}
deploy:
resources:
reservations:
devices:
- driver: nvidia
# You must change these IDs depending on the server
# e.g., if you use DGX series: `0, 1, 2, 4` to avoid using crappy GPU
# e.g., otherwise: 0, 1, 2, 3
device_ids: ['0', '1', '2', '3']
capabilities: [gpu]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment