查看GPU的占用进程

原创已于 2023-02-03 17:21:36 修改 · 847 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#linux

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

程序调试专栏收录该内容

1 篇文章

订阅专栏

使用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