Spaces:
Runtime error
Runtime error
Commit
·
473c15c
1
Parent(s):
03e01a8
update dockerfile
Browse files- dockerfile +11 -3
dockerfile
CHANGED
|
@@ -1,13 +1,18 @@
|
|
| 1 |
-
|
| 2 |
FROM nvcr.io/nvidia/pytorch:23.08
|
| 3 |
|
| 4 |
LABEL maintainer="yuanze"
|
| 5 |
LABEL email="[email protected]"
|
| 6 |
|
|
|
|
|
|
|
| 7 |
# Install webp support
|
| 8 |
RUN apt update && apt install libwebp-dev -y
|
| 9 |
|
| 10 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
|
| 12 |
# note that you may need to modify the TORCH_CUDA_ARCH_LIST in the setup.py file
|
| 13 |
ENV TORCH_CUDA_ARCH_LIST="8.6"
|
|
@@ -16,4 +21,7 @@ ENV TORCH_CUDA_ARCH_LIST="8.6"
|
|
| 16 |
RUN git clone https://github.com/erikwijmans/Pointnet2_PyTorch.git \
|
| 17 |
&& cp -f change_setup.txt Pointnet2_PyTorch/pointnet2_ops_lib/setup.py \
|
| 18 |
&& cd Pointnet2_PyTorch/pointnet2_ops_lib \
|
| 19 |
-
&& pip install .
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
FROM nvcr.io/nvidia/pytorch:23.08
|
| 2 |
|
| 3 |
LABEL maintainer="yuanze"
|
| 4 |
LABEL email="[email protected]"
|
| 5 |
|
| 6 |
+
WORKDIR /code
|
| 7 |
+
|
| 8 |
# Install webp support
|
| 9 |
RUN apt update && apt install libwebp-dev -y
|
| 10 |
|
| 11 |
+
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"
|
|
|
|
| 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"]
|