在Linux下按内存限制进程

时间:2013-11-28 16:04:04

标签: linux memory process

我们必须在Linux系统上启动几个饥饿的进程。这些过程通常需要运行几个Go(~5Go)内存(总内存:16Go RAM + 2Go交换)。

  • 首先,当系统内存不足时,OOM杀手会终止进程,每次发生时我们都必须重新启动系统。

  • 然后,我们尝试使用overcommit_memory(= 2)+ overcommit_ratio(= 75)参数,因此在情况变得严重时不会启动进程。因此,无需重新启动服务器。但是,我们的启动脚本现在会在达到限制时报告几十个错误:新进程立即出错,进程从未启动。

  • 所以现在我们正在寻找一个解决方案来启动“尽可能多”的流程,然后他们会被延迟/暂停或等待他们的兄弟停止......是否存在?< / p>

1 个答案:

答案 0 :(得分:0)

我编写了这个小工具,允许并行运行任意数量的作业,或者限制并行度。只需阅读标题中的注释即可查看如何使用它 - 您可以将作业数量限制为2个或3个或类似的内容,但可以一次性提交。它在下面使用REDIS,它是免费的,并且非常容易安装。

#!/bin/bash
################################################################################
# File: core
# Author: Mark Setchell
# 
# Primitive, but effective tool for managing parallel execution of jobs in the
# shell. Based on, and requiring REDIS.
#
# Usage:
#
# core -i 8 # Initialise to 8 cores, or specify 0 to use all available cores
# for i in {0..63}
# do
#   # Wait for a core, do a process, release core
#   (core -p; process; core -v)&
# done
# wait
################################################################################
function usage {
    echo "Usage: core -i ncores # Initialise with ncores. Use 0 for all cores."
    echo "       core -p        # Wait (forever) for free core."
    echo "       core -v        # Release core."
    exit 1
}

function init {
    # Delete list of cores in REDIS
    echo DEL cores | redis-cli > /dev/null 2>&1
    for i in `seq 1 $NCORES`
    do
       # Add another core to list of cores in REDIS
       echo LPUSH cores 1 | redis-cli > /dev/null 2>&1
    done
    exit 0
}

function WaitForCore {
    # Wait forever for a core to be available
    echo BLPOP cores 0 | redis-cli > /dev/null 2>&1
    exit 0
}

function ReleaseCore {
    # Release or give back a core
    echo LPUSH cores 1 | redis-cli > /dev/null 2>&1
    exit 0
}

################################################################################
# Main
################################################################################
while getopts "i:pv" optname
  do
    case "$optname" in
      "i")
        if [ $OPTARG -lt 1 ]; then
           NCORES=`sysctl -n hw.logicalcpu`;    # May differ if not on OSX
        else
           NCORES=$OPTARG
        fi
    init $NCORES
        ;;
      "p")
    WaitForCore
        ;;
      "v")
    ReleaseCore
        ;;
      "?")
        echo "Unknown option $OPTARG"
        ;;
    esac
done
usage