From 697c20d97791d5a7fbbdfaa26824f0c98b985e50 Mon Sep 17 00:00:00 2001 From: Pedro Larroy Date: Wed, 3 Apr 2019 02:43:07 +0000 Subject: [PATCH] Fix dockerized GPU builds in dev_menu --- dev_menu.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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',