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="yuanze1024@gmail.com"
|
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="yuanze1024@gmail.com"
|
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"]
|