Update docker files to TensorFlow 1, Theano 0.9 (#6116)

- TensorFlow 1
- Theano 0.9 : also use "device=cuda" in theanorc to use new
"gpuarray" backend
- Miniconda 4.2.12 (latest conda installer with python 3.5)
- Simplified pip install for tensorflow and keras test dependencies
Esse commit está contido em:
Durgesh Mankekar
2017-04-03 11:33:41 -04:00
commit de François Chollet
commit b943176d2a
2 arquivos alterados com 12 adições e 11 exclusões
+11 -10
Ver Arquivo
@@ -7,10 +7,10 @@ RUN mkdir -p $CONDA_DIR && \
echo export PATH=$CONDA_DIR/bin:'$PATH' > /etc/profile.d/conda.sh && \ echo export PATH=$CONDA_DIR/bin:'$PATH' > /etc/profile.d/conda.sh && \
apt-get update && \ apt-get update && \
apt-get install -y wget git libhdf5-dev g++ graphviz && \ apt-get install -y wget git libhdf5-dev g++ graphviz && \
wget --quiet https://repo.continuum.io/miniconda/Miniconda3-3.9.1-Linux-x86_64.sh && \ wget --quiet https://repo.continuum.io/miniconda/Miniconda3-4.2.12-Linux-x86_64.sh && \
echo "6c6b44acdd0bc4229377ee10d52c8ac6160c336d9cdd669db7371aa9344e1ac3 *Miniconda3-3.9.1-Linux-x86_64.sh" | sha256sum -c - && \ echo "c59b3dd3cad550ac7596e0d599b91e75d88826db132e4146030ef471bb434e9a *Miniconda3-4.2.12-Linux-x86_64.sh" | sha256sum -c - && \
/bin/bash /Miniconda3-3.9.1-Linux-x86_64.sh -f -b -p $CONDA_DIR && \ /bin/bash /Miniconda3-4.2.12-Linux-x86_64.sh -f -b -p $CONDA_DIR && \
rm Miniconda3-3.9.1-Linux-x86_64.sh rm Miniconda3-4.2.12-Linux-x86_64.sh
ENV NB_USER keras ENV NB_USER keras
ENV NB_UID 1000 ENV NB_UID 1000
@@ -24,13 +24,14 @@ RUN useradd -m -s /bin/bash -N -u $NB_UID $NB_USER && \
USER keras USER keras
# Python # Python
ARG python_version=3.5.2 ARG python_version=3.5
ARG tensorflow_version=0.12.0rc0-cp35-cp35m
RUN conda install -y python=${python_version} && \ RUN conda install -y python=${python_version} && \
pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-${tensorflow_version}-linux_x86_64.whl && \ pip install --upgrade pip && \
pip install git+git://github.com/Theano/Theano.git && \ pip install tensorflow-gpu && \
pip install ipdb pytest pytest-cov python-coveralls coverage==3.7.1 pytest-xdist pep8 pytest-pep8 pydot_ng && \ conda install Pillow scikit-learn notebook pandas matplotlib mkl nose pyyaml six h5py && \
conda install Pillow scikit-learn notebook pandas matplotlib nose pyyaml six h5py && \ conda install theano pygpu && \
git clone git://github.com/fchollet/keras.git /src && pip install -e /src[tests] && \
pip install git+git://github.com/fchollet/keras.git && \ pip install git+git://github.com/fchollet/keras.git && \
conda clean -yt conda clean -yt
+1 -1
Ver Arquivo
@@ -1,5 +1,5 @@
[global] [global]
floatX = float32 floatX = float32
optimizer=None optimizer=None
device = gpu device = cuda