FROM nvidia/cuda:9.2-cudnn7-runtime-ubuntu18.04
RUN apt-get update --fix-missing && \
apt-get install -y --no-install-recommends \
wget bzip2 ca-certificates curl git build-essential \
cmake python3.7 python3.7-dev python3.7-distutils pkg-config && \
apt-get clean && \
rm -rf /var/lib/apt/lists/*
RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py && python3.7 get-pip.py
# We're emulating nvidia-smi to be able to build GPU container on devices without GPU
RUN echo '#!/bin/bash\necho Building container with GPU support' > /usr/bin/nvidia-smi && \
chmod +x /usr/bin/nvidia-smi
ADD . /gluonts
RUN pip3.7 install /gluonts[shell]
ENTRYPOINT ["python3.7", "-m", "gluonts.shell"]