Skip to content

Commit 370b0ee

Browse files
committed
add the dockerfile
1 parent bddc411 commit 370b0ee

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed

Dockerfile

+31
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
FROM pytorch/pytorch:2.4.0-cuda12.4-cudnn9-devel
2+
3+
ENV DEBIAN_FRONTEND=noninteractive
4+
5+
# update system
6+
RUN apt-get update && apt-get upgrade -y
7+
8+
# install build-essential
9+
RUN apt install -y \
10+
build-essential \
11+
git \
12+
libgsl-dev \
13+
libx11-6 \
14+
libgl1-mesa-glx \
15+
libglib2.0-0 \
16+
libsm6 \
17+
libxext6 \
18+
libxrender-dev
19+
20+
# install requirements
21+
RUN pip install wandb==0.16.6 open3d==0.18.0 optuna==3.6.1 trimesh==4.4.3 numpy==2.0.1
22+
23+
# copy the code
24+
COPY . /workspace
25+
WORKDIR /workspace
26+
27+
# build ND-Net
28+
WORKDIR /workspace/core_legacy/build
29+
RUN cmake ..
30+
RUN make -j8
31+
RUN cp ./libndnet.so /usr/local/lib/libndnet.so

0 commit comments

Comments
 (0)