add requirments

e0d1779f · LDOUBLEV · f3efa9be · e0d1779f · e0d1779f
隐藏空白更改
内联并排

Showing with 10 addition and 0 deletion

test/infer.sh test/infer.sh +5 -0

test/test.sh test/test.sh +5 -0

未找到文件。
--- 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

--- 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