FROM pytorch/pytorch

RUN apt-get update && apt-get install -y git

RUN pip install git+https://github.com/google-research/torchsde.git@v0.2.4
RUN pip install matplotlib
RUN pip install pandas
RUN pip install gpytorch==1.4.2
ADD . .

