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 |
|