查看GPU的占用进程

Yupeng5619

已于 2023-02-03 17:21:36 修改

阅读量564

点赞数

分类专栏：程序调试文章标签： linux

于 2023-02-03 14:49:34 首次发布

本文链接：https://blog.csdn.net/qq_42327424/article/details/128866908

版权

程序调试专栏收录该内容

1 篇文章 0 订阅

订阅专栏

使用ps

ps -aux | grep $PATTERN

可以查看执行的命令和pid
2. 使用

fuser -v /dev/nvidia*"

删除所有占用gpu的进程

fuser -kv /dev/nvidia*"

根据指令的模糊匹配kill

#!/bin/bash

declare -a pids
declare -a cmds

a="$(ps -ax | grep $1)"
oldIFS=$IFS
IFS=$'\n'
cnt=0
for line in $a; do
    IFS=$oldIFS read -a arr <<< $line
    pids[$cnt]=${arr[0]}
    cmds[$cnt]=${arr[-1]}
    (( cnt+=1 ))
done

(( len=${#pids[@]}-3 ))
lineidx=0
if [ $len -eq 0 ] ; then
    echo No process found.
else
    while [ $lineidx -lt $len ]; do
        echo kill process, [command]: ${cmds[$lineidx]} [pid]: ${pids[$lineidx]}
        kill -9 ${pids[$lineidx]}
        (( lineidx+=1 ))
    done
fi