FROM nvcr.io/nvidia/pytorch:23.08

LABEL maintainer="yuanze"
LABEL email="yuanze1024@gmail.com"

# Install webp support
RUN apt update && apt install libwebp-dev -y

RUN pip install -r requirements.txt

# note that you may need to modify the TORCH_CUDA_ARCH_LIST in the setup.py file
ENV TORCH_CUDA_ARCH_LIST="8.6"

# Install Pointnet2_PyTorch
RUN git clone https://github.com/erikwijmans/Pointnet2_PyTorch.git \
    && mv -f backup_install.txt Pointnet2_PyTorch/pointnet2_ops_lib/setup.py \
    && cd Pointnet2_PyTorch/pointnet2_ops_lib \
    && python install .