maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

[MRCNN-PYT] Update opencv-python version, fix numpy version

Browse source
This commit is contained in:
Shriya Balaji Palsamudram 2021-11-04 14:13:48 -07:00
parent 2592d5a02c
commit 53f95e5a5f
4 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
PyTorch/Segmentation/MaskRCNN/Dockerfile
View file

@ -18,7 +18,7 @@ FROM ${FROM_IMAGE_NAME}
# Install Python dependencies
RUN pip install --no-cache-dir \
mlperf-compliance==0.0.10 \
opencv-python==4.2.0.32 \
opencv-python==4.4.0.42 \
git+https://github.com/NVIDIA/dllogger \
yacs
@ -29,6 +29,6 @@ RUN mkdir /results
COPY pytorch/. .
RUN pip install -e .
RUN /opt/conda/bin/conda install -y numpy==1.17.4
RUN /opt/conda/bin/conda install -y numpy
ENV OMP_NUM_THREADS=1

2
PyTorch/Segmentation/MaskRCNN/README.md
View file

@ -597,7 +597,7 @@ To achieve these same results, follow the steps in the [Quick Start Guide](#quic
October 2021
- Replace APEX AMP with PyTorch native AMP
- Use opencv-python version 4.2.0.32
- Use opencv-python version 4.4.0.42
June 2020
- Updated accuracy and performance tables to include A100 results

4
PyTorch/Segmentation/MaskRCNN/pytorch/Dockerfile
View file

@ -18,7 +18,7 @@ FROM ${FROM_IMAGE_NAME}
# Install Python dependencies
RUN pip install --no-cache-dir \
mlperf-compliance==0.0.10 \
opencv-python==4.2.0.32 \
opencv-python==4.4.0.42 \
git+https://github.com/NVIDIA/dllogger \
yacs
@ -29,6 +29,6 @@ RUN mkdir /results
COPY . .
RUN pip install -e .
RUN /opt/conda/bin/conda install -y numpy==1.17.4
RUN /opt/conda/bin/conda install -y numpy
ENV OMP_NUM_THREADS=1

2
PyTorch/Segmentation/MaskRCNN/pytorch/requirements.txt
View file

@ -1,4 +1,4 @@
mlperf-compliance==0.0.10
opencv-python==4.2.0.32
opencv-python==4.4.0.42
yacs
git+https://github.com/NVIDIA/cocoapi.git@nvidia/master#egg=cocoapi&subdirectory=PythonAPI