FROM python

RUN apt-get update
RUN pip install numpy
RUN pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html

RUN apt-get install -y git
RUN git clone https://github.com/stealthBanana/CrypTen.git
WORKDIR /CrypTen
RUN git checkout issue_454
RUN pip install -e .