add requirments

This commit is contained in:
LDOUBLEV 2021-06-09 11:27:36 +00:00
parent f3efa9be6b
commit e0d1779fae
2 changed files with 10 additions and 0 deletions

View File

@ -29,6 +29,11 @@ infer_gpu_id=$(func_parser "${lines[12]}")
log_path=$(func_parser "${lines[13]}") log_path=$(func_parser "${lines[13]}")
status_log="${log_path}/result.log" 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(){ function status_check(){
last_status=$1 # the exit code last_status=$1 # the exit code

View File

@ -69,6 +69,11 @@ gpu_precision_list=$(func_parser "${lines[11]}")
log_path=$(func_parser "${lines[13]}") log_path=$(func_parser "${lines[13]}")
status_log="${log_path}/result.log" 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(){ function status_check(){
last_status=$1 # the exit code last_status=$1 # the exit code
run_model=$2 run_model=$2