diff --git a/model_zoo/official/cv/googlenet/scripts/run_train_gpu.sh b/model_zoo/official/cv/googlenet/scripts/run_train_gpu.sh
index 3f2419f7eb..b30160238d 100644
--- a/model_zoo/official/cv/googlenet/scripts/run_train_gpu.sh
+++ b/model_zoo/official/cv/googlenet/scripts/run_train_gpu.sh
@@ -41,5 +41,11 @@ mkdir ../train
 cd ../train || exit
 
 export CUDA_VISIBLE_DEVICES="$2"
-mpirun -n $1 --allow-run-as-root \
-python3 ${BASEPATH}/../train.py > train.log 2>&1 &
+
+if [ $1 -gt 1 ]
+then
+    mpirun -n $1 --allow-run-as-root \
+    python3 ${BASEPATH}/../train.py > train.log 2>&1 &
+else
+    python3 ${BASEPATH}/../train.py > train.log 2>&1 &
+fi