diff --git a/docker/Dockerfile.demo_cpu b/docker/Dockerfile.demo_cpu
index 6700579bc41b..95a1dbce100b 100644
--- a/docker/Dockerfile.demo_cpu
+++ b/docker/Dockerfile.demo_cpu
@@ -17,7 +17,7 @@
 
 # Minimum docker image for demo purposes
 # prebuilt-image: tvmai/demo-cpu
-FROM tvmai/ci-cpu:v0.52
+FROM tvmai/ci-cpu:v0.54
 
 # Jupyter notebook.
 RUN pip3 install matplotlib Image Pillow jupyter[notebook]
diff --git a/docker/install/install_tvm_cpu.sh b/docker/install/install_tvm_cpu.sh
index d013ad2b5d4e..a2719e82c150 100755
--- a/docker/install/install_tvm_cpu.sh
+++ b/docker/install/install_tvm_cpu.sh
@@ -21,7 +21,7 @@ set -u
 set -o pipefail
 
 cd /usr
-git clone --depth=1 https://github.com/apache/incubator-tvm tvm --recursive
+git clone https://github.com/apache/incubator-tvm tvm --recursive
 cd /usr/tvm
 # checkout a hash-tag
 git checkout 4b13bf668edc7099b38d463e5db94ebc96c80470
diff --git a/docker/install/install_tvm_gpu.sh b/docker/install/install_tvm_gpu.sh
index 8798e4a303d9..d9b9a0a183d1 100755
--- a/docker/install/install_tvm_gpu.sh
+++ b/docker/install/install_tvm_gpu.sh
@@ -21,7 +21,7 @@ set -u
 set -o pipefail
 
 cd /usr
-git clone --depth=1 https://github.com/apache/incubator-tvm tvm --recursive
+git clone https://github.com/apache/incubator-tvm tvm --recursive
 cd /usr/tvm
 # checkout a hash-tag
 git checkout 4b13bf668edc7099b38d463e5db94ebc96c80470