diff --git a/test/test.sh b/test/test.sh index 1cbb36012b5be45ad9ecf718df4df4d6f165c76f..f2ac3f8b29af1be08e8eb5b836133dc53ad3d5b2 100644 --- a/test/test.sh +++ b/test/test.sh @@ -219,6 +219,7 @@ for gpu in ${gpu_list[*]}; do eval $env save_infer_path="${save_log}" func_inference "${python}" "${inference_py}" "${save_infer_path}" "${LOG_PATH}" "${infer_img_dir}" + eval "unset CUDA_VISIBLE_DEVICES" done done done