[MRCNN-PYT] Update opencv-python version, fix numpy version
This commit is contained in:
parent
2592d5a02c
commit
53f95e5a5f
|
@ -18,7 +18,7 @@ FROM ${FROM_IMAGE_NAME}
|
||||||
# Install Python dependencies
|
# Install Python dependencies
|
||||||
RUN pip install --no-cache-dir \
|
RUN pip install --no-cache-dir \
|
||||||
mlperf-compliance==0.0.10 \
|
mlperf-compliance==0.0.10 \
|
||||||
opencv-python==4.2.0.32 \
|
opencv-python==4.4.0.42 \
|
||||||
git+https://github.com/NVIDIA/dllogger \
|
git+https://github.com/NVIDIA/dllogger \
|
||||||
yacs
|
yacs
|
||||||
|
|
||||||
|
@ -29,6 +29,6 @@ RUN mkdir /results
|
||||||
COPY pytorch/. .
|
COPY pytorch/. .
|
||||||
RUN pip install -e .
|
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
|
ENV OMP_NUM_THREADS=1
|
||||||
|
|
|
@ -597,7 +597,7 @@ To achieve these same results, follow the steps in the [Quick Start Guide](#quic
|
||||||
|
|
||||||
October 2021
|
October 2021
|
||||||
- Replace APEX AMP with PyTorch native AMP
|
- 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
|
June 2020
|
||||||
- Updated accuracy and performance tables to include A100 results
|
- Updated accuracy and performance tables to include A100 results
|
||||||
|
|
|
@ -18,7 +18,7 @@ FROM ${FROM_IMAGE_NAME}
|
||||||
# Install Python dependencies
|
# Install Python dependencies
|
||||||
RUN pip install --no-cache-dir \
|
RUN pip install --no-cache-dir \
|
||||||
mlperf-compliance==0.0.10 \
|
mlperf-compliance==0.0.10 \
|
||||||
opencv-python==4.2.0.32 \
|
opencv-python==4.4.0.42 \
|
||||||
git+https://github.com/NVIDIA/dllogger \
|
git+https://github.com/NVIDIA/dllogger \
|
||||||
yacs
|
yacs
|
||||||
|
|
||||||
|
@ -29,6 +29,6 @@ RUN mkdir /results
|
||||||
COPY . .
|
COPY . .
|
||||||
RUN pip install -e .
|
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
|
ENV OMP_NUM_THREADS=1
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
mlperf-compliance==0.0.10
|
mlperf-compliance==0.0.10
|
||||||
opencv-python==4.2.0.32
|
opencv-python==4.4.0.42
|
||||||
yacs
|
yacs
|
||||||
git+https://github.com/NVIDIA/cocoapi.git@nvidia/master#egg=cocoapi&subdirectory=PythonAPI
|
git+https://github.com/NVIDIA/cocoapi.git@nvidia/master#egg=cocoapi&subdirectory=PythonAPI
|
||||||
|
|
Loading…
Reference in a new issue