Commit 8a657534 authored by zhaoying1's avatar zhaoying1
Browse files

Update run_train.sh

parent 73613eea
...@@ -18,8 +18,7 @@ hostfile=./hostfile ...@@ -18,8 +18,7 @@ hostfile=./hostfile
np=$(cat $hostfile|sort|uniq |wc -l) np=$(cat $hostfile|sort|uniq |wc -l)
np=$(($np*8)) np=$(($np*8))
nodename=$(cat $hostfile |sed -n "1p")
dist_url=`echo $nodename | awk '{print $1}'`
which mpirun which mpirun
mpirun -np $np --allow-run-as-root --hostfile hostfile --bind-to none --mca btl_tcp_if_include $dist_url run_train_single.sh mpirun -np $np --allow-run-as-root --hostfile hostfile --bind-to none --mca btl_tcp_if_include enp97s0f1 run_train_single.sh 8
echo "END TIME: $(date)" echo "END TIME: $(date)"
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment