Add ability to customize DOCKER_GPU_OPTS env variable to handle weird…

… handling of multiple GPUs by Docker (#1576) Signed-off-by: Navin Kumar <[email protected]>
NVIDIA · Nov 21, 2023 · 0eec2c6 · 0eec2c6
1 parent 592e007
commit 0eec2c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/build/run-in-docker b/build/run-in-docker
@@ -42,7 +42,7 @@ $DOCKER_CMD build $DOCKER_BUILD_EXTRA_ARGS -f $REPODIR/ci/Dockerfile \
   $REPODIR/build
 
 if [[ "$DOCKER_CMD" == "docker" ]]; then
-  DOCKER_GPU_OPTS="--gpus all"
+  DOCKER_GPU_OPTS=${DOCKER_GPU_OPTS:-"--gpus all"}
 fi
 
 if (( $# == 0 )); then