FROM r200-8f_xmlir-ubuntu_2004_x86_64:v0.27

WORKDIR /root

SHELL ["/bin/bash", "-c"]

ADD xpytorch-cp38-torch201-ubuntu2004-x64-socket.run /root/
RUN source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda && \
    ./xpytorch-cp38-torch201-ubuntu2004-x64-socket.run

WORKDIR /workspace
COPY run_diffuser_test_r200_8f.py dataset.json /workspace/