diff --git a/deployment/Dockerfile.build.base b/deployment/Dockerfile.build.base
index 56315a3b5f..465cb77034 100644
--- a/deployment/Dockerfile.build.base
+++ b/deployment/Dockerfile.build.base
@@ -64,7 +64,7 @@ RUN wget -qO- http://archive.apache.org/dist/hadoop/common/hadoop-${HADOOP_VERSI
 #
 #Install NNI
 #
-RUN pip3 install -v --user git+https://github.com/Microsoft/nni.git@v0.2
+RUN pip3 install -v --user git+https://github.com/Microsoft/nni.git@$(curl --silent "https://api.github.com/repos/Microsoft/nni/releases/latest" | grep '"tag_name":' | sed -E 's/.*"([^"]+)".*/\1/')
 
 ENV JAVA_HOME=/usr/lib/jvm/java-8-openjdk-amd64 \
     HADOOP_INSTALL=/usr/local/hadoop \