Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2019-09-09 10:20:03 +02:00
parent 7706a1b8fb
commit 48a0f38f85
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 # conda install pytorch torchvision -c pytorch
# Install OpenCV with Gstreamer support # Install OpenCV with Gstreamer support
WORKDIR /usr/src #WORKDIR /usr/src
RUN pip uninstall -y opencv-python #RUN pip uninstall -y opencv-python
RUN apt-get update #RUN apt-get update
RUN apt-get install -y gstreamer1.0-python3-dbg-plugin-loader #RUN apt-get install -y gstreamer1.0-python3-dbg-plugin-loader
# RUN apt-get install gstreamer1.0 ## RUN apt-get install gstreamer1.0
# RUN apt install -y ubuntu-restricted-extras ## RUN apt install -y ubuntu-restricted-extras
RUN apt install -y libgstreamer1.0-dev #RUN apt install -y libgstreamer1.0-dev
RUN apt install -y libgstreamer-plugins-base1.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.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 git clone https://github.com/opencv/opencv_contrib.git && cd opencv_contrib && git checkout 4.1.1
RUN cd opencv/build && cmake ../ \ #RUN cd opencv/build && cmake ../ \
-D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules \ # -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules \
-D BUILD_OPENCV_PYTHON3=ON \ # -D BUILD_OPENCV_PYTHON3=ON \
-D PYTHON3_EXECUTABLE=/opt/conda/bin/python \ # -D PYTHON3_EXECUTABLE=/opt/conda/bin/python \
-D PYTHON3_INCLUDE_PATH=/opt/conda/include/python3.6m \ # -D PYTHON3_INCLUDE_PATH=/opt/conda/include/python3.6m \
-D PYTHON3_LIBRARIES=/opt/conda/lib/python3.6/site-packages \ # -D PYTHON3_LIBRARIES=/opt/conda/lib/python3.6/site-packages \
-D WITH_GSTREAMER=ON \ # -D WITH_GSTREAMER=ON \
-D WITH_FFMPEG=OFF \ # -D WITH_FFMPEG=OFF \
&& make && make install && ldconfig # && 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 /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 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())" #RUN python3 -c "import cv2; print(cv2.getBuildInformation())"
# Create working directory # Create working directory
RUN mkdir -p /usr/src/app RUN mkdir -p /usr/src/app