Allocate extra space in /dev/shm to enable NCCL

dmlc · Feb 12, 2023 · 4e9e6c5 · 4e9e6c5
1 parent 7127abd
commit 4e9e6c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/buildkite/test-cpp-mgpu.sh b/tests/buildkite/test-cpp-mgpu.sh
@@ -4,7 +4,8 @@ set -euo pipefail
 
 source tests/buildkite/conftest.sh
 
-export CI_DOCKER_EXTRA_PARAMS_INIT='-e NCCL_DEBUG=INFO'
+# Allocate extra space in /dev/shm to enable NCCL
+export CI_DOCKER_EXTRA_PARAMS_INIT='--shm-size=4g'
 
 echo "--- Run Google Tests with CUDA, using multiple GPUs"
 buildkite-agent artifact download "build/testxgboost" . --step build-cuda