如何在Docker容器内显式设置应用可见的内存量

问题原因解释 #

Docker通过cgroup实现内存限制（你配置的deploy.resources.limits.memory: 3gb是生效的，容器内存占用超过3GB时会被OOM终止），但容器内的/proc/meminfo默认是从宿主机挂载的虚拟文件系统，所以显示的是宿主机的总内存，而非容器的内存限制。这是Docker的默认行为，不影响内存限制的实际生效，但会误导依赖/proc/meminfo做内存判断的应用（比如你的Python程序）。

解决方案 #

方案1：修改Python应用的内存检测逻辑（推荐） #

直接让Python读取cgroup的内存限制文件来获取真实可用内存，而非依赖/proc/meminfo。根据cgroup版本不同，读取对应的文件：

• cgroup v1：读取/sys/fs/cgroup/memory/memory.limit_in_bytes
• cgroup v2：读取/sys/fs/cgroup/memory.max

示例Python代码片段：

def get_container_memory_limit():
    try:
        # 先尝试cgroup v2
        with open('/sys/fs/cgroup/memory.max', 'r') as f:
            limit = f.read().strip()
            if limit != 'max':
                return int(limit) // (1024 * 1024)  # 转换为MB
        # 再尝试cgroup v1
        with open('/sys/fs/cgroup/memory/memory.limit_in_bytes', 'r') as f:
            limit = int(f.read().strip())
            return limit // (1024 * 1024)
    except FileNotFoundError:
        # fallback到宿主机内存（非容器环境）
        with open('/proc/meminfo', 'r') as f:
            for line in f:
                if line.startswith('MemTotal'):
                    return int(line.split()[1]) // 1024
    return None

方案2：修改容器内/proc/meminfo的显示 #

如果必须让/proc/meminfo显示容器的内存限制，可以在容器启动时动态替换MemTotal行的内容。在docker-compose.yml中修改command字段，先执行替换操作再启动Python应用：

services:
  your-python-service:
    # 其他配置（镜像、端口等）
    deploy:
      resources:
        limits:
          memory: 3gb
    command: >
      sh -c '
        # 获取cgroup内存限制（转换为KB）
        if [ -f /sys/fs/cgroup/memory.max ]; then
          MEM_LIMIT=$(cat /sys/fs/cgroup/memory.max)
        else
          MEM_LIMIT=$(cat /sys/fs/cgroup/memory/memory.limit_in_bytes)
        fi
        MEM_TOTAL=$((MEM_LIMIT / 1024))
        # 替换/proc/meminfo中的MemTotal行
        sed -i "s/^MemTotal:.*/MemTotal:       $MEM_TOTAL kB/" /proc/meminfo
        # 启动Python应用
        exec python your_app.py
      '

注意：部分系统中/proc是只读虚拟文件系统，这种方法可能无法生效，此时优先选择方案1。

验证内存限制是否生效 #

可以在容器内执行以下命令验证cgroup限制：

# cgroup v1
cat /sys/fs/cgroup/memory/memory.limit_in_bytes
# cgroup v2
cat /sys/fs/cgroup/memory.max

返回值应为3221225472（即3GB对应的字节数），说明内存限制已正确应用。

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