Docker Compose 可以通过以下方式控制 CPU 内存使用和 GPU 显存使用:
- 控制 CPU 内存使用:可以在 docker-compose.yml 文件中指定
cpu_shares和mem_limit属性来限制容器的 CPU 使用率和内存使用量。例如:
version: '3'
services:
my_service:
image: my_image
cpu_shares: 512
mem_limit: 500m
上面的示例将限制容器最多只能使用 512 的 CPU 资源,并且最多只能使用 500MB 的内存。
- 控制 GPU 显存使用:如果您正在运行基于 TensorFlow、PyTorch 或其他深度学习框架的应用程序,则可以通过设置环境变量或挂载 CUDA 设备来控制容器的 GPU 显存使用。例如:
version: '3'
services:
my_service:
image: my_image
environment:
- NVIDIA_VISIBLE_DEVICES=all
- NVIDIA_DRIVER_CAPABILITIES=all
- TF_FORCE_GPU_ALLOW_GROWTH=true
volumes:
- /usr/local/cuda:/usr/local/cuda
上面的示例将允许容器访问所有可用的 NVIDIA 显卡设备,并挂载 CUDA 库,同时还启用了 TensorFlow 的自适应 GPU 内存分配模式。




