File size: 551 Bytes
4e9ddf9
 
 
 
 
b5a071f
1
2
3
4
5
6
7
# The first item builds our docker image on each node once.
# The second item spawns one docker container per GPU.
# {{ gpu }} is defined in `hosts.yaml`, and will be filled in when Pegasus
# determines the specific node and gpu the generated job command will run on.
# We check {{ gpu }} = 0 to ensure that the image is only built once on each node.
- docker run -dit --name leaderboard{{ gpu }} --gpus '"device={{ gpu }}"' -v /data/leaderboard:/data/leaderboard -v $HOME/workspace/leaderboard:/workspace/leaderboard mlenergy/leaderboard:latest bash