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
Show Stats 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 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 # Install Tensorflow
RUN pip install --quiet \ RUN conda install --quiet --yes \
'tensorflow==2.1.0' && \ 'tensorflow=1.13*' \
'keras=2.2*' && \
conda clean --all -f -y && \
fix-permissions $CONDA_DIR && \ fix-permissions $CONDA_DIR && \
fix-permissions /home/$NB_USER fix-permissions /home/$NB_USER
############################################################################
########################## Dependency: pytorch #############################
############################################################################
# Install PyTorch dependencies # Install PyTorch dependencies
RUN conda install --quiet --yes \ RUN conda install --quiet --yes \
pyyaml mkl mkl-include setuptools cmake cffi typing && \ pyyaml mkl mkl-include setuptools cmake cffi typing && \

10
generate_Dockerfile.sh
View File

@ -53,21 +53,15 @@ echo "
" >> $DOCKERFILE " >> $DOCKERFILE
cat $STACKS_DIR/datascience-notebook/Dockerfile | grep -v BASE_CONTAINER >> $DOCKERFILE cat $STACKS_DIR/datascience-notebook/Dockerfile | grep -v BASE_CONTAINER >> $DOCKERFILE
echo "
############################################################################
################ Dependency: jupyter/tensorflow-notebook ###################
############################################################################
" >> $DOCKERFILE
cat $STACKS_DIR/tensorflow-notebook/Dockerfile | grep -v BASE_CONTAINER >> $DOCKERFILE
# Note that the following step also installs the cudatoolkit, which is # Note that the following step also installs the cudatoolkit, which is
# essential to access the GPU. # essential to access the GPU.
echo " echo "
############################################################################ ############################################################################
########################## Dependency: pytorch ############################# ########################## Dependency: gpulibs #############################
############################################################################ ############################################################################
" >> $DOCKERFILE " >> $DOCKERFILE
cat src/Dockerfile.pytorch >> $DOCKERFILE cat src/Dockerfile.gpulibs >> $DOCKERFILE
echo " echo "

8
src/Dockerfile.pytorch → src/Dockerfile.gpulibs
View File

@ -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 # Install PyTorch dependencies
RUN conda install --quiet --yes \ RUN conda install --quiet --yes \
pyyaml mkl mkl-include setuptools cmake cffi typing && \ pyyaml mkl mkl-include setuptools cmake cffi typing && \