run_1gpu.sh 258 Bytes
Newer Older
liangjing's avatar
liangjing committed
1
2
3
4
5
6
7
#!/bin/bash
cp rundir_1gpu/init_env.py .
export PADDLE_TRAINERS_NUM=1
export PADDLE_TRAINER_ENDPOINTS=localhost:60001
export SEED=${SEED:-"$RANDOM"}

mpirun -np 1 --allow-run-as-root -x PADDLE_TRAINERS_NUM -x PADDLE_TRAINER_ENDPOINTS ./run_benchmark_1gpu.sh