如何在Python中获取容器内存限制？

Question

我正在尝试使用 Python 检索为其中的 Docker 容器设置的实际内存限制：

docker run --rm -it --memory="2g" python:3.8 python -c "import os; print((os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES'))/(1024.**3))"

但它返回主机的所有可用内存。

我知道我可以使用 Python 的 Docker 包和绑定安装

/var/run/docker.sock

从检查容器配置中获取该信息，但我需要知道是否还有其他方法，因为我无法使用该方法。

Answer 1

容器的内存限制是由linux cgroups控制的，因此您可以获取容器中

/sys/fs/cgroup/memory/memory.limit_in_bytes

的值来计算容器内的限制内存：

root@pie:~# docker run --rm -it --memory="2g" python:3.8 /bin/bash
root@e22c4275f26c:/# python3
Python 3.8.15 (default, Oct 14 2022, 00:19:58)
[GCC 10.2.1 20210110] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import subprocess
>>> limit_in_bytes=subprocess.check_output(["cat", "/sys/fs/cgroup/memory/memory.limit_in_bytes"]).decode("utf-8")
>>> print(int(limit_in_bytes)/(1024**3))
2.0
>>>

Answer 2

vasya@fbi:~$ docker run -it --rm -u1000:1000 -m 128m --memory-reservation 64m python:3.11
Python 3.11.8 (main, Mar 12 2024, 11:41:52) [GCC 12.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from pathlib import Path
>>> Path("/sys/fs/cgroup/memory.low").read_text()
'67108864\n'
>>> Path("/sys/fs/cgroup/memory.max").read_text()
'134217728\n'
>>>