Linux命令行删除当前用户未使用的共享内存段的实现方法

直接可用的单行命令 #

执行以下命令即可完成清理，不会影响其他用户资源和正在使用的共享内存：

ipcs -m | awk -v user="$(whoami)" '$3 == user && $6 == 0 && $7 == "dest" {print $2}' | xargs -r ipcrm -m

各部分逻辑说明：

• -v user="$(whoami)"：将当前系统用户名传入awk作为变量
• $3 == user：仅过滤当前用户所有的共享内存段
• $6 == 0：确认共享内存段的附着进程数为0，无程序正在使用
• $7 == "dest"：匹配状态为dest（已标记待销毁）的段
• print $2：输出符合条件的共享内存ID（shmid）
• xargs -r：如果没有符合条件的待清理段，就不执行后续删除命令，避免报错
• ipcrm -m <shmid>：系统提供的删除指定共享内存段的原生命令

可复用的shell脚本版本 #

将以下内容保存为cleanup_shm.sh，执行chmod +x cleanup_shm.sh赋予执行权限后，即可在程序运行结束后直接调用：

#!/bin/bash
# 清理当前用户下状态为dest的闲置共享内存段

CURRENT_USER=$(whoami)
# 筛选符合条件的共享内存ID
SHM_LIST=$(ipcs -m | awk -v user="$CURRENT_USER" '$3 == user && $6 == 0 && $7 == "dest" {print $2}')

if [ -z "$SHM_LIST" ]; then
    echo "无符合条件的待清理共享内存段"
    exit 0
fi

echo "即将清理以下闲置共享内存段ID："
echo "$SHM_LIST"
# 执行删除操作
echo "$SHM_LIST" | xargs -r ipcrm -m

if [ $? -eq 0 ]; then
    echo "清理完成"
else
    echo "清理过程出现错误"
    exit 1
fi

注意事项 #

• 清理逻辑仅会匹配当前用户所有、无进程附着、状态为dest的共享内存段，不会影响Pytorch等正在运行的任务占用的共享内存（正在使用的段附着进程数大于0，状态也不会标记为dest）
• 可以先执行ipcs -m | awk -v user="$(whoami)" '$3 == user && $6 == 0 && $7 == "dest" {print $2}'预览待清理的共享内存ID，确认无误后再执行删除操作
• 部分发行版ipcs输出列顺序如果有差异，可先手动执行ipcs -m核对列顺序，调整awk对应的列号即可

内容的提问来源于stack exchange，提问作者user2454869