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:
commit de
François Chollet
pai
f9c9c0ab3f
commit
b943176d2a
+11
-10
@@ -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
@@ -1,5 +1,5 @@
|
|||||||
[global]
|
[global]
|
||||||
floatX = float32
|
floatX = float32
|
||||||
optimizer=None
|
optimizer=None
|
||||||
device = gpu
|
device = cuda
|
||||||
|
|
||||||
|
|||||||
Referência em uma Nova Issue
Bloquear um usuário