# CUDA9, cudnn7のdocker imageを基本にする
FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04

# /rootをワークディレクトリにする
WORKDIR /root
ENV HOME /root

# 入力待ちをオフにする
ENV DEBIAN_FRONTEND noninteractive

# apt-get
RUN apt-get update && \
apt-get install -y --no-install-recommends \
wget \
unzip \
make \
git \
libssl-dev \
libbz2-dev \
libsqlite3-dev \
libreadline-dev \
zlib1g-dev
# pyenv
RUN git clone https://github.com/yyuu/pyenv.git ~/.pyenv
ENV PYENV_ROOT $HOME/.pyenv
ENV PATH $PYENV_ROOT/shims:$PYENV_ROOT/bin:$PATH
RUN pyenv install 3.6.6
RUN pyenv global 3.6.6

# pip
RUN pip install --no-cache-dir --upgrade pip && \
pip install --no-cache-dir \
tensorflow-gpu