Hello_User/gpu-jupyter
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

gpu-libs installer instead of splitted tensorflow and pytorch

Browse Source
This commit is contained in:
Christoph Schranz
2020-02-24 21:35:15 +01:00
parent a1aab35da1
commit f8848a73b4
3 changed files with 15 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
.build/Dockerfile
View File

@@ -353,24 +353,17 @@ RUN julia -e 'import Pkg; Pkg.update()' && \
fix-permissions $JULIA_PKGDIR $CONDA_DIR/share/jupyter
############################################################################
################ Dependency: jupyter/tensorflow-notebook ###################
########################## Dependency: gpulibs #############################
############################################################################
# Copyright (c) Jupyter Development Team.
# Distributed under the terms of the Modified BSD License.
LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"
# Install Tensorflow
RUN pip install --quiet \
'tensorflow==2.1.0' && \
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
############################################################################
########################## Dependency: pytorch #############################
############################################################################
# Install PyTorch dependencies
RUN conda install --quiet --yes \
pyyaml mkl mkl-include setuptools cmake cffi typing && \