Deep Learningの環境を構築するためのDockerfile

Dockerfile

FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu16.04
 
RUN apt-get update
RUN apt-get install -y git wget curl emacs
# pyenv
ENV PYENV_ROOT $HOME/.pyenv
ENV PATH $PYENV_ROOT/shims:$PYENV_ROOT/bin:$PATH
ENV PYTHON_VERSION anaconda3-4.2.0
 
RUN git clone https://github.com/pyenv/pyenv.git $PYENV_ROOT
RUN pyenv init -
RUN pyenv install $PYTHON_VERSION
RUN pyenv rehash
RUN pyenv global $PYTHON_VERSION
RUN pip install --upgrade pip
# chainer
ENV LD_LIBRARY_PATH $LD_LIBRARY_PATH:/usr/local/cuda/lib64/:/usr/local/cuda/lib64/stubs
RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1
RUN pip install cupy
RUN pip install chainer
# tensorflow
RUN pip install tensorflow-gpu
# keras
RUN pip install keras
# pytorch
RUN conda install pytorch torchvision cuda80 -c soumith
# other python libraries
RUN pip install bcolz
 
# gdrive
RUN curl -JLO https://docs.google.com/uc?id=0B3X9GlR6EmbnQ0FtZmJJUXEyRTA&export=download
RUN chmod +x gdrive-linux-x64
RUN mv gdrive-linux-x64 /usr/local/bin
RUN echo 'alias gdrive=gdrive-linux-x64' >> ~/.bashrc
# alias
RUN echo 'alias ll="ls -l"' >> ~/.bashrc
RUN echo 'alias b="cd .."' >> ~/.bashrc
ENTRYPOINT ["/bin/bash"]