# 定义你要清理的卡号
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