├── keras.json
├── README.md
├── Dockerfile
└── LICENSE


/keras.json:
--------------------------------------------------------------------------------
1 | {
2 |     "epsilon": 1e-07,
3 |     "backend": "tensorflow",
4 |     "floatx": "float32",
5 |     "image_dim_ordering": "tf"
6 | }


--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
 1 | # tf-keras-cuda-docker
 2 | 
 3 | Docker image for Tensorflow and Keras with CUDA support
 4 | 
 5 | Requires [nvidia-docker](https://github.com/NVIDIA/nvidia-docker)
 6 | 
 7 | - CUDA: 8.0
 8 | - cuDNN: 5
 9 | - Python: 3.5
10 | - Tensorflow: 0.12.0rc0
11 | - Keras: 1.1.2
12 | 
13 | Additional libs:
14 | - ipython
15 | - numpy
16 | - scipy
17 | - pandas
18 | - pillow
19 | - scikit-image
20 | - scikit-learn
21 | - pydicom
22 | - h5py
23 | 


--------------------------------------------------------------------------------
/Dockerfile:
--------------------------------------------------------------------------------
 1 | FROM nvidia/cuda:8.0-cudnn5-devel
 2 | MAINTAINER Leon Chen <leon@md.ai>
 3 | 
 4 | RUN apt-get update -y && apt-get install -y git wget
 5 | 
 6 | RUN wget https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3.sh && \
 7 |     bash ~/miniconda3.sh -b -p $HOME/miniconda3 && \
 8 |     rm ~/miniconda3.sh
 9 | 
10 | ENV PATH "$HOME/miniconda3/bin:$PATH"
11 | 
12 | RUN pip install --upgrade pip
13 | RUN conda install -y \
14 |     ipython \
15 |     numpy \
16 |     scipy \
17 |     pandas \
18 |     pillow \
19 |     scikit-image \
20 |     scikit-learn \
21 |     h5py
22 | 
23 | RUN git clone https://github.com/darcymason/pydicom.git && \
24 |     cd pydicom && \
25 |     python setup.py install && \
26 |     cd .. && \
27 |     rm -rf pydicom
28 | 
29 | ENV TENSORFLOW_VERSION 0.12.0rc0
30 | ENV KERAS_VERSION 1.1.2
31 | RUN pip --no-cache-dir install \
32 |     https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-${TENSORFLOW_VERSION}-cp35-cp35m-linux_x86_64.whl
33 | RUN pip install keras==${KERAS_VERSION}
34 | 
35 | COPY keras.json $HOME/.keras/keras.json
36 | 


--------------------------------------------------------------------------------
/LICENSE:
--------------------------------------------------------------------------------
 1 | MIT License
 2 | 
 3 | Copyright (c) 2016 Leon Chen
 4 | 
 5 | Permission is hereby granted, free of charge, to any person obtaining a copy
 6 | of this software and associated documentation files (the "Software"), to deal
 7 | in the Software without restriction, including without limitation the rights
 8 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 9 | copies of the Software, and to permit persons to whom the Software is
10 | furnished to do so, subject to the following conditions:
11 | 
12 | The above copyright notice and this permission notice shall be included in all
13 | copies or substantial portions of the Software.
14 | 
15 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21 | SOFTWARE.
22 | 


--------------------------------------------------------------------------------