gpu-libs installer instead of splitted tensorflow and pytorch
This commit is contained in:
parent
a1aab35da1
commit
f8848a73b4
@ -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 && \
|
||||
|
@ -53,21 +53,15 @@ echo "
|
||||
" >> $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
|
||||
# essential to access the GPU.
|
||||
echo "
|
||||
############################################################################
|
||||
########################## Dependency: pytorch #############################
|
||||
########################## Dependency: gpulibs #############################
|
||||
############################################################################
|
||||
" >> $DOCKERFILE
|
||||
cat src/Dockerfile.pytorch >> $DOCKERFILE
|
||||
cat src/Dockerfile.gpulibs >> $DOCKERFILE
|
||||
|
||||
|
||||
echo "
|
||||
|
@ -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 && \
|
Loading…
Reference in New Issue
Block a user