diff --git a/docker/dockerfile.tf b/docker/dockerfile.tf
index 803fcf1e3..2568d9af7 100644
--- a/docker/dockerfile.tf
+++ b/docker/dockerfile.tf
@@ -59,9 +59,11 @@ RUN if [ "$INSTALL_NVT" == "true" ]; then \
     fi
 
 # Install Transformers4Rec
-RUN git clone https://github.com/NVIDIA-Merlin/Transformers4Rec.git /transformers4rec && \
-    cd /transformers4rec/;  if [ "$RELEASE" == "true" ] && [ ${TF4REC_VER} != "vnightly" ] ; then git fetch --all --tags && git checkout tags/${TF4REC_VER}; else git checkout main; fi; \
-    pip install -e .[tensorflow,nvtabular] && python setup.py develop
+RUN if [ "$INSTALL_NVT" == "true" ]; then \
+        git clone https://github.com/NVIDIA-Merlin/Transformers4Rec.git /transformers4rec && \
+        cd /transformers4rec/;  if [ "$RELEASE" == "true" ] && [ ${TF4REC_VER} != "vnightly" ] ; then git fetch --all --tags && git checkout tags/${TF4REC_VER}; else git checkout main; fi; \
+        pip install -e .[tensorflow,nvtabular] && python setup.py develop; \
+    fi
 
 # Install HugeCTR
 ENV LD_LIBRARY_PATH=/usr/local/hugectr/lib:$LD_LIBRARY_PATH \