pytorch3d

Dockerfile

@@ -66,9 +66,12 @@ SHELL ["conda", "run","--no-capture-output", "-p","/app/env", "/bin/bash", "-c"]
 
 RUN conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia && \
     conda clean -ya && \
-    conda install -c "nvidia/label/cuda-12.1.0" cuda-nvcc && conda clean -ya
+    conda install -c "nvidia/label/cuda-12.1.0" cuda-nvcc && conda clean -ya \
+    conda install pytorch3d -c pytorch3d && conda clean -ya
 # conda install -c "nvidia/label/cuda-12.1.0" cuda-toolkit && conda clean -ya
 
+
+
 COPY --chown=1000:1000 . /app/
 RUN make socket-kit.so