Spaces:
Runtime error
Runtime error
File size: 732 Bytes
38293b0 f15a1cd 473c15c f15a1cd 473c15c f15a1cd ff4d2cd 473c15c ff4d2cd 473c15c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
FROM nvcr.io/nvidia/pytorch:23.08-py3
LABEL maintainer="yuanze"
LABEL email="yuanze1024@gmail.com"
WORKDIR /code
# Install webp support
RUN apt update && apt install libwebp-dev -y
COPY ./requirements.txt /code/requirements.txt
RUN pip install --no-cache-dir --upgrade -r /code/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
# replace the setup.py file to fix the error
COPY ./change_setup.txt /code/Pointnet2_PyTorch/pointnet2_ops_lib/setup.py
RUN cd Pointnet2_PyTorch/pointnet2_ops_lib \
&& pip install .
COPY . .
CMD ["python", "app.py"] |