在我使用的 HPC 集群上,我希望将环境变量与 docker 容器一起使用,以将容器 GPU 设备链接到分配的 GPU。
docker run -it ${DOCKER_GPU_ARGUMENTS} --rm image bash
花费到:
docker run -it --memory=64g --memory-swap=64g --gpus all --device=/dev/nvidia-uvm --device=/dev/nvidia-uvm-tools --device=/dev/nvidiactl --device /dev/nvidia0 --rm image bash
有没有办法在 docker-compose 中使用该环境变量?
是的,请阅读 deploy 属性
的文档