[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
|
||||
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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in a new issue