Update Dockerfile

Dockerfile

@@ -51,7 +51,8 @@ RUN curl -sLo ~/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-late
  && chmod +x ~/miniconda.sh \
  && ~/miniconda.sh -b -p ~/miniconda \
  && rm ~/miniconda.sh \
- && conda clean -ya
+ && conda install -y python=3.6 && \
+    conda clean -ya
 
 WORKDIR $HOME/app
 
@@ -79,9 +80,8 @@ RUN mkdir /data && chown user:user /data
 
 USER user
 
-# Install TensorFlow 1.15 in Conda
-RUN pip install --no-cache-dir tensorflow-gpu==1.15
-
+# Install TensorFlow 1.15 using Conda
+RUN conda install -c conda-forge tensorflow-gpu=1.15
 
 # Python packages
 RUN --mount=target=requirements.txt,source=requirements.txt \