docker-gpu.df

FROM nvidia/cuda:7.5-cudnn4-devel

# Install dependencies
RUN apt-get -qq update            && \
    apt-get -qq install --assume-yes \
        "module-init-tools"         \
        "build-essential"           \
        "cmake"                     \
        "git"                       \
        "wget"                      \
        "libopenjpeg2"              \
        "libopenblas-dev"           \
        "liblapack-dev"             \
        "libjpeg-dev"               \
        "libtiff5-dev"              \
        "zlib1g-dev"                \
        "libfreetype6-dev"          \
        "liblcms2-dev"              \
        "libwebp-dev"               \
        "gfortran"                  \
        "pkg-config"		    \
        "python3"                   \
        "python3-dev"               \
        "python3-pip"               \
        "python3-numpy"             \
        "python3-scipy"             \
        "python3-matplotlib"        \
        "python3-six"               \
        "python3-networkx"          \
        "python3-tk"             &&  \
    rm -rf /var/lib/apt/lists/*  &&  \
    python3 -m pip -q install "cython"

# Install requirements before copying project files
WORKDIR /nd
COPY requirements.txt .
RUN python3 -m pip -q install -r "requirements.txt"

# Copy only required project files
COPY doodle.py .

# Get a pre-trained neural network, non-commercial & attribution. (GELU2)
RUN wget -q "https://github.com/alexjc/neural-doodle/releases/download/v0.0/gelu2_conv.pkl"

# Set an entrypoint to the main doodle.py script
ENTRYPOINT ["python3", "doodle.py", "--device=gpu"]