diff --git a/Dockerfile-runtime b/Dockerfile-runtime index d13681928..1e7b0ce36 100644 --- a/Dockerfile-runtime +++ b/Dockerfile-runtime @@ -1,10 +1,12 @@ ARG docker_name -ARG cudaversion FROM $docker_name MAINTAINER H2o.ai +# Refer https://github.com/moby/moby/issues/34129 for weird ARG behavior +ARG cudaversion + ENV HOME=/root ENV CUDA_HOME=/usr/local/cuda ENV PATH=/usr/local/cuda/bin:$PATH