gpu-libs installer instead of splitted tensorflow and pytorch

2020-02-24 21:35:15 +01:00
parent a1aab35da1
commit f8848a73b4
3 changed files with 15 additions and 20 deletions
--- a/src/Dockerfile.gpulibs
+++ b/src/Dockerfile.gpulibs
@@ -1,3 +1,11 @@
+# Install Tensorflow
+RUN conda install --quiet --yes \
+    'tensorflow=1.13*' \
+    'keras=2.2*' && \
+    conda clean --all -f -y && \
+    fix-permissions $CONDA_DIR && \
+    fix-permissions /home/$NB_USER
+
 # Install PyTorch dependencies
 RUN conda install --quiet --yes \
    pyyaml mkl mkl-include setuptools cmake cffi typing && \