Fix dockerized GPU builds in dev_menu (apache#14603)

haohuanw · Jun 23, 2019 · 7a4d70a · 7a4d70a
1 parent ee5e17e
commit 7a4d70a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dev_menu.py b/dev_menu.py
@@ -132,12 +132,12 @@ def create_virtualenv_default():
     ]),
     ('[Docker] Python3 GPU unittests',
     [
-        "ci/build.py --platform ubuntu_gpu /work/runtime_functions.sh build_ubuntu_gpu",
+        "ci/build.py --nvidiadocker --platform ubuntu_gpu /work/runtime_functions.sh build_ubuntu_gpu",
         "ci/build.py --nvidiadocker --platform ubuntu_gpu /work/runtime_functions.sh unittest_ubuntu_python3_gpu",
     ]),
     ('[Docker] Python3 GPU+MKLDNN unittests',
     [
-        "ci/build.py --platform ubuntu_gpu /work/runtime_functions.sh build_ubuntu_gpu_cmake_mkldnn",
+        "ci/build.py --nvidiadocker --platform ubuntu_gpu /work/runtime_functions.sh build_ubuntu_gpu_cmake_mkldnn",
         "ci/build.py --nvidiadocker --platform ubuntu_gpu /work/runtime_functions.sh unittest_ubuntu_python3_gpu",
     ]),
     ('[Docker] Python3 CPU Intel MKLDNN unittests',