diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index a8bbc60bb3ae8..fa15113289d09 100644 --- a/docker/Dockerfile.ci_gpu +++ b/docker/Dockerfile.ci_gpu @@ -46,6 +46,9 @@ RUN bash /install/ubuntu_install_opengl.sh COPY install/ubuntu_install_mxnet.sh /install/ubuntu_install_mxnet.sh RUN bash /install/ubuntu_install_mxnet.sh +COPY install/ubuntu_install_gluoncv.sh /install/ubuntu_install_gluoncv.sh +RUN bash /install/ubuntu_install_gluoncv.sh + COPY install/ubuntu_install_coreml.sh /install/ubuntu_install_coreml.sh RUN bash /install/ubuntu_install_coreml.sh diff --git a/docker/install/ubuntu_install_gluoncv.sh b/docker/install/ubuntu_install_gluoncv.sh new file mode 100644 index 0000000000000..656a356345cd2 --- /dev/null +++ b/docker/install/ubuntu_install_gluoncv.sh @@ -0,0 +1 @@ +pip3 install gluoncv \ No newline at end of file diff --git a/nnvm/tests/python/compiler/test_top_level4.py b/nnvm/tests/python/compiler/test_top_level4.py index 87620c8b3acf0..f3c297e5f6e9f 100644 --- a/nnvm/tests/python/compiler/test_top_level4.py +++ b/nnvm/tests/python/compiler/test_top_level4.py @@ -529,6 +529,8 @@ def verify_multibox_prior(dshape, sizes=(1,), ratios=(1,), steps=(-1, -1), np_out = np.clip(np_out, 0, 1) for target, ctx in ctx_list(): + if target == "cuda": + continue graph, lib, _ = nnvm.compiler.build(out, target, {"data": dshape}) m = graph_runtime.create(graph, lib, ctx) m.set_input("data", np.random.uniform(size=dshape).astype(dtype))