yuanze1024 commited on
Commit
ff4d2cd
1 Parent(s): 38293b0

Add application file

Browse files
Files changed (2) hide show
  1. .dockerignore +4 -0
  2. Dockerfile +7 -5
.dockerignore ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ .cache
2
+ __pycache__
3
+ *.pt
4
+ *.log
Dockerfile CHANGED
@@ -12,16 +12,18 @@ COPY ./requirements.txt /code/requirements.txt
12
 
13
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
14
 
15
- COPY . .
16
-
17
  # note that you may need to modify the TORCH_CUDA_ARCH_LIST in the setup.py file
18
  ENV TORCH_CUDA_ARCH_LIST="8.6"
19
 
20
  # Install Pointnet2_PyTorch
21
- RUN git clone https://github.com/erikwijmans/Pointnet2_PyTorch.git \
22
- && cp -f change_setup.txt Pointnet2_PyTorch/pointnet2_ops_lib/setup.py \
23
- && cd Pointnet2_PyTorch/pointnet2_ops_lib \
 
 
 
24
  && pip install .
25
 
 
26
 
27
  CMD ["python", "app.py"]
 
12
 
13
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
14
 
 
 
15
  # note that you may need to modify the TORCH_CUDA_ARCH_LIST in the setup.py file
16
  ENV TORCH_CUDA_ARCH_LIST="8.6"
17
 
18
  # Install Pointnet2_PyTorch
19
+ RUN git clone https://github.com/erikwijmans/Pointnet2_PyTorch.git
20
+
21
+ # replace the setup.py file to fix the error
22
+ COPY ./change_setup.txt /code/Pointnet2_PyTorch/pointnet2_ops_lib/setup.py
23
+
24
+ RUN cd Pointnet2_PyTorch/pointnet2_ops_lib \
25
  && pip install .
26
 
27
+ COPY . .
28
 
29
  CMD ["python", "app.py"]