yuanze1024 commited on
Commit
473c15c
1 Parent(s): 03e01a8

update dockerfile

Browse files
Files changed (1) hide show
  1. 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
- RUN pip install -r requirements.txt
 
 
 
 
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"]