Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2019-09-09 10:19:46 +02:00
parent b4b93be693
commit 7706a1b8fb
1 changed files with 22 additions and 22 deletions

View File

@ -9,28 +9,28 @@ FROM nvcr.io/nvidia/pytorch:19.08-py3
# conda install pytorch torchvision -c pytorch
# Install OpenCV with Gstreamer support
#WORKDIR /usr/src
#RUN pip uninstall -y opencv-python
#RUN apt-get update
#RUN apt-get install -y gstreamer1.0-python3-dbg-plugin-loader
## RUN apt-get install gstreamer1.0
## RUN apt install -y ubuntu-restricted-extras
#RUN apt install -y libgstreamer1.0-dev
#RUN apt install -y libgstreamer-plugins-base1.0-dev
#RUN git clone https://github.com/opencv/opencv.git && cd opencv && git checkout 4.1.1 && mkdir build
#RUN git clone https://github.com/opencv/opencv_contrib.git && cd opencv_contrib && git checkout 4.1.1
#RUN cd opencv/build && cmake ../ \
# -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules \
# -D BUILD_OPENCV_PYTHON3=ON \
# -D PYTHON3_EXECUTABLE=/opt/conda/bin/python \
# -D PYTHON3_INCLUDE_PATH=/opt/conda/include/python3.6m \
# -D PYTHON3_LIBRARIES=/opt/conda/lib/python3.6/site-packages \
# -D WITH_GSTREAMER=ON \
# -D WITH_FFMPEG=OFF \
# && make && make install && ldconfig
#RUN cd /usr/local/lib/python3.6/site-packages/ && mv cv2.cpython-36m-x86_64-linux-gnu.so cv2.so
#RUN cd /opt/conda/lib/python3.6/site-packages/ && ln -s /usr/local/lib/python3.6/site-packages/cv2.so cv2.so
#RUN python3 -c "import cv2; print(cv2.getBuildInformation())"
WORKDIR /usr/src
RUN pip uninstall -y opencv-python
RUN apt-get update
RUN apt-get install -y gstreamer1.0-python3-dbg-plugin-loader
# RUN apt-get install gstreamer1.0
# RUN apt install -y ubuntu-restricted-extras
RUN apt install -y libgstreamer1.0-dev
RUN apt install -y libgstreamer-plugins-base1.0-dev
RUN git clone https://github.com/opencv/opencv.git && cd opencv && git checkout 4.1.1 && mkdir build
RUN git clone https://github.com/opencv/opencv_contrib.git && cd opencv_contrib && git checkout 4.1.1
RUN cd opencv/build && cmake ../ \
-D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules \
-D BUILD_OPENCV_PYTHON3=ON \
-D PYTHON3_EXECUTABLE=/opt/conda/bin/python \
-D PYTHON3_INCLUDE_PATH=/opt/conda/include/python3.6m \
-D PYTHON3_LIBRARIES=/opt/conda/lib/python3.6/site-packages \
-D WITH_GSTREAMER=ON \
-D WITH_FFMPEG=OFF \
&& make && make install && ldconfig
RUN cd /usr/local/lib/python3.6/site-packages/cv2/python-3.6/ && mv cv2.cpython-36m-x86_64-linux-gnu.so cv2.so
RUN cd /opt/conda/lib/python3.6/site-packages/ && ln -s /usr/local/lib/python3.6/site-packages/cv2/python-3.6/cv2.so cv2.so
RUN python3 -c "import cv2; print(cv2.getBuildInformation())"
# Create working directory
RUN mkdir -p /usr/src/app