Dockerfile

FROM nvidia/cuda:9.0-cudnn7-runtime
MAINTAINER CodaLab Team "codalab.worksheets@gmail.com"

# Begin common steps (Must be the same in the CPU and GPU images)
RUN apt-get update && apt-get install -y \
    apt-transport-https \
    iputils-ping \
    git \
    python2.7 \
    python-pip \
    python-dev \
    python-software-properties \
    python-tk \
    software-properties-common \
    build-essential \
    cmake \
    libhdf5-dev \
    swig \
    wget \
    curl

## Python 3.6
RUN add-apt-repository ppa:deadsnakes/ppa && \
    apt-get update -y  && \
    apt-get install python3.7 -y \
        python3.7-venv \
        python3.7-dev \
        python3-software-properties

RUN curl https://bootstrap.pypa.io/get-pip.py | python3.7

### Without this Python thinks we're ASCII and unicode chars fail
ENV LANG C.UTF-8

## Oracle JDK 8
RUN echo oracle-java8-installer shared/accepted-oracle-license-v1-1 select true | debconf-set-selections && \
    add-apt-repository -y ppa:webupd8team/java && \
    apt-get update && \
    apt-get install -y oracle-java8-installer && \
    rm -rf /var/cache/oracle-jdk8-installer

ENV JAVA_HOME /usr/lib/jvm/java-8-oracle
ENV SCALA_VERSION 2.12.6

## Scala
RUN wget http://scala-lang.org/files/archive/scala-$SCALA_VERSION.deb && \
    dpkg -i scala-$SCALA_VERSION.deb && \
    echo "deb https://dl.bintray.com/sbt/debian /" | tee -a /etc/apt/sources.list.d/sbt.list && \
    apt-key adv --keyserver hkp://keyserver.ubuntu.com:80 --recv 2EE0EA64E40A89B84B2DF73499E82A75642AC823 && \
    apt-get -y update && \
    apt-get -y install sbt && \
    apt-get clean && \
    apt-get autoremove && \
    rm -rf /var/lib/apt/lists/*

ENV SCALA_HOME /usr/share/java

RUN pip2 install -U pip
RUN pip2 install -U \
      numpy \
      scipy \
      matplotlib \
      pandas

## Set up python3.7 environment

RUN pip3 install -U pip
RUN pip3 install -U \
      numpy \
      scipy \
      matplotlib \
      pandas \
      sympy \
      nose \
      tqdm \
      wheel \
      scikit-learn \
      scikit-image \
      crayons \
      pendulum \
      six \
      toml \
      yapf

# End common steps
# GPU-specific commands

RUN pip3 install -U \
      torch \
      torchvision