监听进程结束并自动运行新进程与虚拟环境智能切换的sh脚本

文章目录

前言
一、等待进程结束自动运行新进程
二、智能环境切换
总结

前言

在编程中，等待进程结束或许是一个很长时间，特别是深度学习。那么这种等待发生在晚上是一个不友好现象，为避免等待情况。我写了一个sh脚本，帮助监听PID，一旦PID进程结束，立即启动我们想运行程序（通常为模型训练），这样我们就不必监控此进程了。当然，深度学习也会涉及自动启动虚拟环境等状况，使其更加智能，我也会在本文中给出sh文件书写方式。

一、等待进程结束自动运行新进程

在这个修改后的Shell脚本中，我们设置了要监听的PID为29455（你可以根据实际情况更改为其他PID），然后在循环中检查该PID是否在运行。如果PID不再运行，则会打印一条消息并立即运行scripts/v1_5/finetune_lora.sh文件。需要说明，如果你执行 scripts/v1_5/finetune_lora.sh此命令需要用什么环境，可提前conda activte “你的环境”，随后类似服务器命令执行方法一样。其sh文件脚本如下：

# 设置要监听的PID
pid_to_monitor=29455while true; doif ! ps -p $pid_to_monitor > /dev/null; thenecho "PID $pid_to_monitor is not running. Starting lora.sh..."sh scripts/v1_5/finetune_lora.shecho "lora.sh started."breakfisleep 5
done

二、智能环境切换

本部分以yolov5的方式来说明，如何切换虚拟环境，如何获得当前路径，如何执行命令等内容。其sh脚本如下：


# train.shecho -e "\n"train time $(date "+%Y-%m-%d")"\n"# 更换虚拟环境__conda_setup="$('/home/anaconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
if [ $? -eq 0 ]; theneval "$__conda_setup"elseif [ -f "/home/anaconda3/etc/profile.d/conda.sh" ]; then. "/home/anaconda3/etc/profile.d/conda.sh"elseexport PATH="/home/anaconda3/bin:$PATH"fi
fi
unset __conda_setup
conda activate yolov5  # 激活环境cur_dir=$(cd `dirname $0`;pwd)  # 获得当前路径
echo -e  "\ncur_dir:"${cur_dir}"\n"yaml_dir=$cur_dir/coco128_hncy.yaml
echo -e  "\nyaml_dir:"${yaml_dir}"\n"save_dir=$cur_dir/runs/train
echo -e "\nsave_dir:"$save_dir"\n"# 判断并构建文件夹
if [ -d ${save_dir} ];thenecho "save_dir 文件存在"elseecho "save_dir文件不存在-->创建文件"mkdir -p  $save_dir
fi# cd 进入模型路径位置
model_dir=/home/Project/yolov5-6.0cd ${model_dir}lsecho -e "\n\n\n\t\t\t start train  ... \n\n\n"python  train.py  --data $yaml_dir   # 执行命令