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 <[email protected]>
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"
