Spaces:
Runtime error
Runtime error
Commit
·
5b2dde1
1
Parent(s):
5b2797c
update dockerfile
Browse files- 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
|
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 |
|