FROM  pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel

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

RUN git clone https://gitea.lnrweb.com/luni/reconstruction.git

RUN conda install -c fvcore -c iopath -c conda-forge fvcore iopath -y
RUN conda install -c bottler nvidiacub -y
RUN conda install pytorch3d -c pytorch3d -y

COPY requirements.txt /tmp/
RUN pip install --requirement /tmp/requirements.txt

WORKDIR /home/workspace