FROM eoelab.org:1027/ben0i0d/jupyter:scipy-c

# install torch
RUN pip install torch torchvision torchaudio -f https://mirror.sjtu.edu.cn/pytorch-wheels/cpu && \
    pip cache purge