HPC集群中如何在Docker Compose使用DOCKER_GPU_ARGUMENTS环境变量？

可以在Docker Compose中使用这个环境变量，但无法直接像docker run那样把环境变量直接插入命令行参数里，需要用以下两种方式变通处理：

方法一：拆解参数到Compose配置字段（推荐） #

把DOCKER_GPU_ARGUMENTS里的各项参数对应到Docker Compose配置文件的指定字段中，既符合Compose规范，也能保证配置的可读性和可维护性。示例配置如下：

version: '3.8'
services:
  gpu-service:
    image: image
    tty: true  # 对应docker run的-it参数
    stdin_open: true
    deploy:
      resources:
        limits:
          memory: ${DOCKER_MEMORY:-64g}  # 从环境变量读取内存限制，默认64g
          memory_swap: ${DOCKER_MEMORY_SWAP:-64g}
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]  # 对应--gpus all参数
    devices:
      # 映射nvidia相关设备文件
      - /dev/nvidia-uvm:/dev/nvidia-uvm
      - /dev/nvidia-uvm-tools:/dev/nvidia-uvm-tools
      - /dev/nvidiactl:/dev/nvidiactl
      - /dev/nvidia0:/dev/nvidia0

如果HPC集群会动态分配GPU设备，还可以把设备路径也改成环境变量，比如：

devices:
  - ${NVIDIA_UVM_DEVICE:-/dev/nvidia-uvm}:/dev/nvidia-uvm
  # 其他设备同理

方法二：通过Shell展开执行Compose命令 #

如果不想修改Compose配置文件，也可以直接在命令行中通过Shell展开环境变量，用docker compose run执行：

docker compose run --rm $(echo ${DOCKER_GPU_ARGUMENTS}) gpu-service bash

这种方式的注意事项：

• 若Compose配置中已存在相关参数（比如内存限制），可能会和环境变量展开的参数冲突；
• 依赖Shell的参数展开功能，不同Shell（如bash、zsh）的处理逻辑可能有差异；
• 如果环境变量中包含特殊字符（如空格、引号），可能会导致参数解析错误。

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