mirror of
https://github.com/deepspeedai/DeepSpeed.git
synced 2025-10-20 15:33:51 +08:00
cd0d6f3c03
* Update scripts to handle cases where you have other VMs in your sub * Support subs with other VMs and fix for PDSH permission error * Minor fix to support subs with other VMs
30 lines
858 B
Bash
Executable File
30 lines
858 B
Bash
Executable File
#!/bin/bash
|
|
|
|
config_file=azure_config.json
|
|
if [ ! -f ${config_file} ]; then
|
|
echo "Cannot find $config_file"
|
|
exit 1
|
|
fi
|
|
|
|
location=`cat ${config_file} | jq .location | sed 's/"//g'`
|
|
rg=deepspeed_rg_$location
|
|
|
|
while getopts 'c:' flag; do
|
|
case "${flag}" in
|
|
c) config_file="${OPTARG}" ;;
|
|
*) error "Unexpected option ${flag}" ;;
|
|
esac
|
|
done
|
|
shift $(expr $OPTIND - 1)
|
|
echo "Using $config_file"
|
|
|
|
nodeid=$1
|
|
cmds=${@:2}
|
|
echo $nodeid $cmds
|
|
ip_addr=`az vm list-ip-addresses -g $rg | jq .[${nodeid}].virtualMachine.network.publicIpAddresses[0].ipAddress | sed 's/"//g'`
|
|
|
|
ssh_private_key=`cat ${config_file} | jq .ssh_private_key | sed 's/"//g'`
|
|
if [ $ssh_private_key == "null" ]; then echo 'missing ssh_private_key in config'; exit 1; fi
|
|
|
|
ssh -i ${ssh_private_key} -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null deepspeed@${ip_addr} ${cmds}
|