diff --git a/tools/run_dali.sh b/tools/run_dali.sh
index 1234f3a83063f08a6d12b8cee41ba0e7909a9eae..4cadc346e821d42da32d05a881b16f3c26579cc7 100755
--- a/tools/run_dali.sh
+++ b/tools/run_dali.sh
@@ -1,5 +1,7 @@
 #!/usr/bin/env bash
 
+export FLAGS_fraction_of_gpu_memory_to_use=0.80
+
 python3.7 -m paddle.distributed.launch \
     --selected_gpus="0,1,2,3" \
     tools/train.py \