Revision 0c892b78737d100b1c9c398f77449568f9fe94b0 authored by Aaron Spieler on 26 February 2020, 15:37:22 UTC, committed by GitHub on 26 February 2020, 15:37:22 UTC
* Added tests to test the gluonts train code of the sagemaker_sdk * Fixing docker images and adding tests to cover some of their functionality. * Marked makefile tests as xfail. Co-Authored-By: Jasper Schulz <jasper.b.schulz@googlemail.com>
1 parent a9c9f7e
Dockerfile.gpu
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"]
Computing file changes ...
