FROM pytorch/pytorch
RUN pip install tensorboard h5py scipy pandas
