Created
April 29, 2018 03:42
-
-
Save Leko/91018ab268bfd8f711d148e68c092fd7 to your computer and use it in GitHub Desktop.
Run Jupyter notebook with Keras for CPU only arch (ex. MacOS)
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
ARG cuda_version=9.0 | |
ARG cudnn_version=7 | |
FROM debian:stretch | |
# Install system packages | |
RUN apt-get update && apt-get install -y --no-install-recommends \ | |
bzip2 \ | |
g++ \ | |
git \ | |
graphviz \ | |
libgl1-mesa-glx \ | |
libhdf5-dev \ | |
openmpi-bin \ | |
wget && \ | |
rm -rf /var/lib/apt/lists/* | |
# Install conda | |
ENV CONDA_DIR /opt/conda | |
ENV PATH $CONDA_DIR/bin:$PATH | |
RUN wget --quiet --no-check-certificate https://repo.continuum.io/miniconda/Miniconda3-4.2.12-Linux-x86_64.sh && \ | |
echo "c59b3dd3cad550ac7596e0d599b91e75d88826db132e4146030ef471bb434e9a *Miniconda3-4.2.12-Linux-x86_64.sh" | sha256sum -c - && \ | |
/bin/bash /Miniconda3-4.2.12-Linux-x86_64.sh -f -b -p $CONDA_DIR && \ | |
rm Miniconda3-4.2.12-Linux-x86_64.sh && \ | |
echo export PATH=$CONDA_DIR/bin:'$PATH' > /etc/profile.d/conda.sh | |
# Install Python packages and keras | |
ENV NB_USER keras | |
ENV NB_UID 1000 | |
RUN useradd -m -s /bin/bash -N -u $NB_UID $NB_USER && \ | |
chown $NB_USER $CONDA_DIR -R && \ | |
mkdir -p /src && \ | |
chown $NB_USER /src | |
USER $NB_USER | |
ARG python_version=3.6 | |
RUN conda install -y python=${python_version} && \ | |
pip install --upgrade pip && \ | |
pip install \ | |
sklearn_pandas \ | |
tensorflow && \ | |
pip install https://cntk.ai/PythonWheel/CPU-Only/cntk-2.1-cp36-cp36m-linux_x86_64.whl && \ | |
conda install \ | |
bcolz \ | |
h5py \ | |
matplotlib \ | |
mkl \ | |
nose \ | |
notebook \ | |
Pillow \ | |
pandas \ | |
pygpu \ | |
pyyaml \ | |
scikit-learn \ | |
six \ | |
theano && \ | |
git clone git://github.com/keras-team/keras.git /src && pip install -e /src[tests] && \ | |
pip install git+git://github.com/keras-team/keras.git && \ | |
conda clean -yt | |
ADD theanorc /home/keras/.theanorc | |
ENV PYTHONPATH='/src/:$PYTHONPATH' | |
WORKDIR /src | |
EXPOSE 8888 | |
CMD jupyter notebook --port=8888 --ip=0.0.0.0 |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
help: | |
@cat Makefile | |
DATA?="${HOME}/Data" | |
GPU?=0 | |
DOCKER_FILE=Dockerfile | |
DOCKER=GPU=$(GPU) docker | |
BACKEND=tensorflow | |
PYTHON_VERSION?=3.6 | |
CUDA_VERSION?=9.0 | |
CUDNN_VERSION?=7 | |
TEST=tests/ | |
SRC?=$(shell dirname `pwd`) | |
build: | |
docker build -t keras --build-arg python_version=$(PYTHON_VERSION) --build-arg cuda_version=$(CUDA_VERSION) --build-arg cudnn_version=$(CUDNN_VERSION) -f $(DOCKER_FILE) . | |
notebook: build | |
$(DOCKER) run -p 8888:8888 -it -v $(SRC):/src/workspace -v $(DATA):/data --env KERAS_BACKEND=$(BACKEND) keras |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment