diff --git a/test/infer.sh b/test/infer.sh
index 324c346bb9f8beb7aa235512bc94eb2de1bd559f..eb70cafbfddbcc87fdc576d88133c410d0abb57a 100644
--- a/test/infer.sh
+++ b/test/infer.sh
@@ -29,6 +29,11 @@ infer_gpu_id=$(func_parser "${lines[12]}")
 log_path=$(func_parser "${lines[13]}")
 status_log="${log_path}/result.log"
 
+# install requirments
+${python} -m pip install pynvml;
+${python} -m pip install psutil;
+${python} -m pip install GPUtil;
+
 
 function status_check(){
     last_status=$1   # the exit code
diff --git a/test/test.sh b/test/test.sh
index cc87c7722fd28bd887ca101284901731842f7474..927abd1913c31649fd6eaf424e95a7871e092c87 100644
--- a/test/test.sh
+++ b/test/test.sh
@@ -69,6 +69,11 @@ gpu_precision_list=$(func_parser "${lines[11]}")
 log_path=$(func_parser "${lines[13]}")
 status_log="${log_path}/result.log"
 
+# install requirments
+${python} -m pip install pynvml;
+${python} -m pip install psutil;
+${python} -m pip install GPUtil;
+
 function status_check(){
     last_status=$1   # the exit code
     run_model=$2