| # 定义你要清理的卡号 | |
| target_gpus="6,7" | |
| # 通过 nvidia-smi 查询，-i 指定卡号，--query 过滤进程 PID | |
| pids=$(nvidia-smi --query-compute-apps=pid --format=csv,noheader -i $target_gpus | sort -u) | |
| if [ -n "$pids" ]; then | |
| echo "正在终止占用 GPU $target_gpus 的进程: $pids" | |
| kill -9 $pids | |
| else | |
| echo "未发现占用 GPU $target_gpus 的进程" | |
| fi |