perseusdg
c0e2097397
- Updated Docker to cuda-11.3+cudnn-8.2.1+TensorRT-8.0.34,Ubuntu to 20.04 and OpenCV to 4.5.4 - Updated OpenCV4 to 4.5.4 in install_OpenCV4.sh - Updated README.md
58 lines
2.0 KiB
Docker
58 lines
2.0 KiB
Docker
FROM nvidia/cuda:11.3.1-devel-ubuntu20.04
|
|
LABEL maintainer "Francesco Gatti"
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
RUN apt-get update && apt-get install libcudnn8-dev=8.2.1.32-1+cuda11.3 libcudnn8=8.2.1.32-1+cuda11.3 libnvinfer-dev=8.0.3-1+cuda11.3 libnvinfer8=8.0.3-1+cuda11.3
|
|
RUN DEBIAN_FRONTEND=noninteractive apt-get update && apt install -y git wget libeigen3-dev libyaml-cpp-dev gcc-9 g++-9 libopengl-dev libgl-dev
|
|
RUN cd /tmp && \
|
|
wget https://github.com/Kitware/CMake/releases/download/v3.21.4/cmake-3.21.4-Linux-x86_64.sh && \
|
|
chmod +x cmake-3.21.4-Linux-x86_64.sh && \
|
|
./cmake-3.21.4-Linux-x86_64.sh --prefix=/usr/local --exclude-subdir --skip-license && \
|
|
rm ./cmake-3.21.4-Linux-x86_64.sh
|
|
ENV NVIDIA_VISIBLE_DEVICES all
|
|
RUN echo "INSTALL OPENCV"
|
|
RUN apt-get install -y build-essential \
|
|
unzip \
|
|
pkg-config \
|
|
libjpeg-dev \
|
|
libpng-dev \
|
|
libtiff-dev \
|
|
libavcodec-dev \
|
|
libavformat-dev \
|
|
libswscale-dev \
|
|
libv4l-dev \
|
|
libxvidcore-dev \
|
|
libx264-dev \
|
|
libgtk-3-dev \
|
|
libatlas-base-dev \
|
|
gfortran-9 \
|
|
libtbb-dev \
|
|
libgstreamer1.0-dev \
|
|
libgstreamer-plugins-base1.0-dev \
|
|
libdc1394-22-dev \
|
|
libavresample-dev
|
|
RUN cd && wget https://github.com/opencv/opencv/archive/4.5.4.tar.gz && tar -xf 4.5.4.tar.gz && rm *.tar.gz
|
|
RUN cd && wget https://github.com/opencv/opencv_contrib/archive/4.5.4.tar.gz && tar -xf 4.5.4.tar.gz && rm *.tar.gz
|
|
RUN cd && \
|
|
cd opencv-4.5.4 && mkdir build && cd build && \
|
|
cmake -D CMAKE_BUILD_TYPE=RELEASE \
|
|
-D CMAKE_INSTALL_PREFIX=/usr/local \
|
|
-D INSTALL_PYTHON_EXAMPLES=OFF \
|
|
-D INSTALL_C_EXAMPLES=OFF \
|
|
-D OPENCV_EXTRA_MODULES_PATH='~/opencv_contrib-4.5.4/modules' \
|
|
-D BUILD_EXAMPLES=OFF \
|
|
-D WITH_CUDA=ON \
|
|
-D CUDA_ARCH_BIN=7.2 \
|
|
-D CUDA_ARCH_PTX="" \
|
|
-D ENABLE_FAST_MATH=ON \
|
|
-D CUDA_FAST_MATH=ON \
|
|
-D WITH_CUBLAS=ON \
|
|
-D WITH_OPENMP=ON \
|
|
-D WITH_LIBV4L=ON \
|
|
-D WITH_GSTREAMER=ON \
|
|
-D WITH_GSTREAMER_0_10=OFF \
|
|
-D WITH_TBB=ON \
|
|
../ && make -j12 && make install
|
|
RUN apt clean
|
|
|
|
|