From 56a99320eb0b91f2f78f1465e65868f047ce7664 Mon Sep 17 00:00:00 2001
From: Ayaz Salikhov <asalikhov@aimtech.team>
Date: Fri, 30 Apr 2021 09:14:04 +0300
Subject: [PATCH] Install tensorflow from conda instead of pip

---
 tensorflow-notebook/Dockerfile | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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}"