From 20f6c84fd2ff3a6f68b62059860cef9cd48661bf Mon Sep 17 00:00:00 2001
From: Changming Sun <chasun@microsoft.com>
Date: Wed, 10 Jul 2019 10:48:42 -0700
Subject: [PATCH] Switch to use nvidia-docker2 command format

---
 tools/ci_build/github/linux/run_dockerbuild.sh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tools/ci_build/github/linux/run_dockerbuild.sh b/tools/ci_build/github/linux/run_dockerbuild.sh
index 8714055cbd..d3e50efd3e 100755
--- a/tools/ci_build/github/linux/run_dockerbuild.sh
+++ b/tools/ci_build/github/linux/run_dockerbuild.sh
@@ -80,13 +80,13 @@ if [ -z "$NIGHTLY_BUILD" ]; then
 fi
 
 if [ $BUILD_DEVICE = "cpu" ] || [ $BUILD_DEVICE = "ngraph" ] || [ $BUILD_DEVICE = "openvino" ]; then
-    ONNX_DOCKER=docker
+    RUNTIME=
 else
-    ONNX_DOCKER=nvidia-docker
+    RUNTIME="--runtime=nvidia"
 fi
 
 docker rm -f "onnxruntime-$BUILD_DEVICE" || true
-$ONNX_DOCKER run -h $HOSTNAME \
+docker run $RUNTIME -h $HOSTNAME \
     --name "onnxruntime-$BUILD_DEVICE" \
     --volume "$SOURCE_ROOT:/onnxruntime_src" \
     --volume "$BUILD_DIR:/build" \