Spaces:
Sleeping
Sleeping
yuanze1024
commited on
Commit
•
ff4d2cd
1
Parent(s):
38293b0
Add application file
Browse files- .dockerignore +4 -0
- 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 |
-
|
23 |
-
|
|
|
|
|
|
|
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"]
|