Files
DeepSpeed/azure/start_container.sh
Jeff Rasley 23a08aebe7 Azure tutorial/scripts (#42)
Add Azure tutorial text and scripts
2020-02-07 17:10:28 -08:00

12 lines
382 B
Bash
Executable File

#!/bin/bash
name=${1-deepspeed}
image=deepspeed/deepspeed:latest
echo "starting docker image named $name"
docker run -d -t --name $name \
--network host \
-v ${HOME}/workdir:/home/deepspeed/workdir \
-v ${HOME}/.ssh:/home/deepspeed/.ssh \
-v /job/hostfile:/job/hostfile \
--gpus all $image bash -c 'sudo service ssh start && sleep infinity'