diff --git a/dev_menu.py b/dev_menu.py
index 1fda4c7aabca..cea0f96793a9 100755
--- 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',