diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh
index 9481f02329cb2800fa372d8523c783faaabc2550..be1849555c1d36e5d8f3bf32e6f94ccbf89a9f48 100644
--- a/test_tipc/test_train_inference_python.sh
+++ b/test_tipc/test_train_inference_python.sh
@@ -247,6 +247,7 @@ else
                 fi
                 # run train
                 eval "unset CUDA_VISIBLE_DEVICES"
+                export FLAGS_cudnn_deterministic=True
                 eval $cmd
                 echo $cmd
                 status_check $? "${cmd}" "${status_log}"