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 <tqc...@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"