Dockerfile for autokeras

Dockerfile

FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04
MAINTAINER Chie HAYASHIDA <chie-hayashida@cookpad.com>

# Install packages
RUN apt-get update && apt-get install -y \
  git \
  vim \
  wget \
  sudo \
  software-properties-common \
  && \
  add-apt-repository ppa:jonathonf/python-3.6 \
  && \
  apt-get update \
  && \
  apt-get install -y \
    build-essential \
    python3.6 \
    python3.6-dev \
    python3.6-venv \
  && \
  wget https://bootstrap.pypa.io/get-pip.py && \
    python3.6 get-pip.py

# settings
ARG user_name=ubuntu
ARG user_id=1000
ARG group_name=ubuntu
ARG group_id=1000

# create user
RUN groupadd -g ${group_id} ${group_name} \
  && \
  useradd -u ${user_id} -g ${group_id} -d /home/${user_name} --create-home --shell /bin/bash ${user_name} \
  && \
  echo "${user_name} ALL=(ALL) NOPASSWD:ALL" >> /etc/sudoers \
  && \
  chown -R ${user_name}:${group_name} /home/${user_name}

# user settings
USER ubuntu
WORKDIR /work
ENV HOME /home/ubuntu
ENV LANG en_US.UTF-8

# Set alias for python3
RUN echo "alias python=python3.6" >> $HOME/.bashrc && \
    echo "alias pip=pip3.6" >> $HOME/.bashrc

COPY ./docker/start.sh /work/start.sh

CMD ["/bin/bash"]