FROM mxnet/python:gpu
RUN pip install keras-mxnet
COPY mnist_cnn.py /tmp/mnist_cnn.py