diff --git a/tensorflow-notebook/Dockerfile b/tensorflow-notebook/Dockerfile
index 2ca96a30..3d9b6919 100644
--- a/tensorflow-notebook/Dockerfile
+++ b/tensorflow-notebook/Dockerfile
@@ -6,7 +6,9 @@ FROM $BASE_CONTAINER
 LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"
 
 # Install Tensorflow
-RUN pip install --quiet --no-cache-dir \
-    'tensorflow==2.4.1' && \
+RUN conda install --quiet --yes \
+    'tensorflow=2.4.1' \
+    && \
+    conda clean --all -f -y && \
     fix-permissions "${CONDA_DIR}" && \
     fix-permissions "/home/${NB_USER}"