yuanze1024 commited on
Commit
5b2dde1
·
1 Parent(s): 5b2797c

update dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -7
Dockerfile CHANGED
@@ -1,3 +1,4 @@
 
1
  FROM nvcr.io/nvidia/pytorch:23.08-py3
2
 
3
  LABEL maintainer="yuanze"
@@ -22,13 +23,7 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
22
  ENV TORCH_CUDA_ARCH_LIST="8.6"
23
 
24
  # Install Pointnet2_PyTorch
25
- RUN git clone https://github.com/erikwijmans/Pointnet2_PyTorch.git
26
-
27
- # replace the setup.py file to fix the error
28
- COPY --chown=user:user ./change_setup.txt /code/Pointnet2_PyTorch/pointnet2_ops_lib/setup.py
29
-
30
- RUN cd Pointnet2_PyTorch/pointnet2_ops_lib \
31
- && pip install .
32
 
33
  COPY --chown=user:user . /code
34
 
 
1
+ # This file may not be used if using the gradio SDK.
2
  FROM nvcr.io/nvidia/pytorch:23.08-py3
3
 
4
  LABEL maintainer="yuanze"
 
23
  ENV TORCH_CUDA_ARCH_LIST="8.6"
24
 
25
  # Install Pointnet2_PyTorch
26
+ RUN pip install "git+git://github.com/yuanze1024/Pointnet2_PyTorch.git#egg=pointnet2_ops&subdirectory=pointnet2_ops_lib"
 
 
 
 
 
 
27
 
28
  COPY --chown=user:user . /code
29