feature: set sparse Dockerfile with Python interpreter only
This commit is contained in:
@@ -1,11 +1,15 @@
|
||||
LABEL maintainer="Christoph Schranz <christoph.schranz@salzburgresearch.at>"
|
||||
|
||||
# Install Tensorflow, check compatibility here: https://www.tensorflow.org/install/gpu
|
||||
RUN conda install --quiet --yes \
|
||||
'tensorflow-gpu=2.1*' \
|
||||
'keras-gpu' && \
|
||||
fix-permissions $CONDA_DIR && \
|
||||
fix-permissions /home/$NB_USER
|
||||
# Install Tensorflow, check compatibility here: https://www.tensorflow.org/install/gpu
|
||||
# installation via conda leads to errors in version 4.8.2
|
||||
#RUN conda install --quiet --yes \
|
||||
# 'tensorflow-gpu=2.1*' \
|
||||
# 'keras-gpu' && \
|
||||
# fix-permissions $CONDA_DIR && \
|
||||
# fix-permissions /home/$NB_USER
|
||||
RUN pip install --upgrade pip && \
|
||||
pip install --no-cache-dir "tensorflow-gpu>=2.1.*" && \
|
||||
pip install --no-cache-dir keras
|
||||
|
||||
# Install PyTorch with dependencies
|
||||
RUN conda install --quiet --yes \
|
||||
|
@@ -2,4 +2,4 @@
|
||||
# The version of cudatoolkit must match those of the base image, see Dockerfile.pytorch
|
||||
FROM nvidia/cuda:10.1-base-ubuntu18.04
|
||||
LABEL maintainer="Christoph Schranz <christoph.schranz@salzburgresearch.at>"
|
||||
# The maintainers of subsequent sections may vary
|
||||
# This is a concatenated Dockerfile, the maintainers of subsequent sections may vary.
|
||||
|
Reference in New Issue
Block a user