Dockerfile

FROM nvidia/cuda:8.0-runtime-ubuntu16.04

RUN apt-get update && \
    apt-get install -y software-properties-common && \
    add-apt-repository universe && \
    add-apt-repository multiverse

RUN apt-get -y update && \
    apt-get -y install python3-pip python3-dev python3-tk vim wget git && \
    pip3 install --upgrade pip && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/* /tmp/*

ENV TERM xterm
ENV ZSH_THEME flazz

RUN apt-get -y update && \
    apt-get install -y zsh && \
    wget https://github.com/robbyrussell/oh-my-zsh/raw/master/tools/install.sh -O - | zsh || true

COPY requirements.txt requirements.txt
RUN pip3 install -r requirements.txt && \
    rm requirements.txt