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