[incubator-tvm] branch master updated: Fix gpu not found when running TVM docker (#4975)

[tq...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git


The following commit(s) were added to refs/heads/master by this push:
     new 7a06bbe  Fix gpu not found when running TVM docker (#4975)
7a06bbe is described below

commit 7a06bbed74eb81c0c1ac267de95b9b5881d13c2f
Author: Tianqi Chen <tq...@users.noreply.github.com>
AuthorDate: Wed Mar 4 18:34:08 2020 -0600

    Fix gpu not found when running TVM docker (#4975)
---
 docker/build.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/build.sh b/docker/build.sh
index a2e21d1..21141b7 100755
--- a/docker/build.sh
+++ b/docker/build.sh
@@ -79,7 +79,7 @@ if [ "$#" -lt 1 ] || [ ! -e "${SCRIPT_DIR}/Dockerfile.${CONTAINER_TYPE}" ]; then
 fi
 
 # Use nvidia-docker if the container is GPU.
-if [[ "${DOCKER_IMAGE_NAME}" == *"gpu"* ]]; then
+if [[ "${CONTAINER_TYPE}" == *"gpu"* ]]; then
     if ! type "nvidia-docker" 1> /dev/null 2> /dev/null
     then
         DOCKER_BINARY="docker"