From e0d1779faee5e79e5c60df75a00348176453cf20 Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Wed, 9 Jun 2021 11:27:36 +0000
Subject: [PATCH] add requirments

---
 test/infer.sh | 5 +++++
 test/test.sh  | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/test/infer.sh b/test/infer.sh
index 324c346b..eb70cafb 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 cc87c772..927abd19 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
-- 
GitLab