FROM tensorflow/tensorflow:latest-gpu 
ENV UNIT=gpu

WORKDIR /ai-benchmark
ADD . /ai-benchmark

RUN pip install .
