Skip to content

Commit 0b3f770

Browse files
ahmadkinv-kkudrynski
ahmadki
authored and
nv-kkudrynski
committed
[SSD/PyT] New release vs 21.05 container
1 parent ffad848 commit 0b3f770

25 files changed

Lines changed: 287 additions & 1584 deletions

‎PyTorch/Detection/SSD/.dockerignore‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
resnet50-19c8e357.pth
2+
nogit/
3+
models/
4+
scripts/
5+

‎PyTorch/Detection/SSD/.gitignore‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,7 @@
11
**/__pycache__
2+
resnet50-19c8e357.pth
3+
nogit/
4+
pbr/
5+
models/
6+
scripts/
7+

‎PyTorch/Detection/SSD/Dockerfile‎

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,20 @@
1-
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:20.06-py3
1+
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:21.05-py3
22
FROM ${FROM_IMAGE_NAME}
33

44
# Set working directory
5-
WORKDIR /workspace
5+
WORKDIR /workspace/ssd
66

7-
ENV PYTHONPATH "${PYTHONPATH}:/workspace"
7+
# Install nv-cocoapi
8+
ENV COCOAPI_VERSION=2.0+nv0.6.0
9+
RUN export COCOAPI_TAG=$(echo ${COCOAPI_VERSION} | sed 's/^.*+n//') \
10+
&& pip install --no-cache-dir pybind11 \
11+
&& pip install --no-cache-dir git+https://github.com/NVIDIA/cocoapi.git@${COCOAPI_TAG}#subdirectory=PythonAPI
12+
# Install dllogger
13+
RUN pip install --no-cache-dir git+https://github.com/NVIDIA/dllogger.git#egg=dllogger
814

15+
# Install requirements
916
COPY requirements.txt .
10-
RUN pip install --no-cache-dir git+https://github.com/NVIDIA/dllogger.git#egg=dllogger
1117
RUN pip install -r requirements.txt
1218
RUN python3 -m pip install pycocotools==2.0.0
1319

14-
# Copy SSD code
15-
COPY ./setup.py .
16-
COPY ./csrc ./csrc
17-
RUN pip install .
18-
1920
COPY . .

‎PyTorch/Detection/SSD/README.md‎

Lines changed: 103 additions & 115 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)