add dockerfile

This commit is contained in:
2022-12-22 22:33:15 +00:00
parent 1dde653df1
commit e846f41178
2 changed files with 19 additions and 0 deletions
+15
View File
@@ -0,0 +1,15 @@
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