Skip to content

Commit a3f7647

Browse files
committed
add dockerfile & update requirements.txt & modify ./lib to enable "make"
1 parent 6f69e46 commit a3f7647

File tree

6 files changed

+271
-256
lines changed

6 files changed

+271
-256
lines changed

Dockerfile

+31
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-devel
2+
3+
ENV TZ=Asia/Seoul
4+
ENV COCOAPI=/workspace/cocoapi
5+
6+
WORKDIR /workspace
7+
8+
RUN apt-get update -y && \
9+
apt-get upgrade -y && \
10+
apt-get install git ffmpeg libsm6 libxext6 -y
11+
12+
RUN mkdir /workspace/deep-high-resolution-net.pytorch
13+
COPY . /workspace/deep-high-resolution-net.pytorch
14+
15+
RUN pip install --no-cache-dir -r requirements.txt
16+
17+
WORKDIR ./deep-high-resolution-net.pytorch/lib
18+
RUN make
19+
20+
RUN git clone https://github.com/cocodataset/cocoapi.git $COCOAPI
21+
WORKDIR $COCOAPI/PythonAPI
22+
RUN make install
23+
RUN python setup.py install --user
24+
25+
ENV NVIDIA_VISIBLE_DEVICES all
26+
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
27+
28+
WORKDIR /workspace/deep-high-resolution-net.pytorch
29+
RUN mkdir output log models data data/coco data/mpii
30+
31+
CMD ["bash", "start.sh"]

0 commit comments

Comments
 (0)