@@ -129,9 +129,9 @@ RUN rm -fr /var/lib/apt/lists/* && \
129
129
ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 && \
130
130
echo "/usr/local/cuda/lib64/stubs" > /etc/ld.so.conf.d/z-cuda-stubs.conf && \
131
131
132
- # Install CUDA-11.3 + cuDNN 8.0.5
132
+ # Install CUDA-11.3 + cuDNN 8.2.0
133
133
ln -s /usr/local/cuda-11.3 /usr/local/cuda && \
134
- ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.8.0.5 /usr/local/cuda/lib64/libcudnn.so && \
134
+ ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.8.2.0 /usr/local/cuda/lib64/libcudnn.so && \
135
135
ldconfig
136
136
137
137
RUN curl -sL https://deb.nodesource.com/setup_14.x | bash - && \
@@ -278,7 +278,7 @@ RUN python3 -m pip install --no-cache-dir -r requirements.py38.txt && \
278
278
rm -f /tmp/*.whl /tmp/requirements.py38.txt
279
279
280
280
# OpenCV
281
- RUN ln -s /usr/include/libv4l1-videodev.h /usr/include/linux/ videodev.h && \
281
+ RUN ln -s /usr/include/libv4l1-videodev.h /usr/include/videodev.h && \
282
282
apt-get install -y \
283
283
libgstreamer1.0-dev \
284
284
libgstreamer-plugins-base1.0-dev \
@@ -315,37 +315,36 @@ RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
315
315
mkdir opencv-${OPENCV_VERSION}/cmake_binary && \
316
316
cd opencv-${OPENCV_VERSION}/cmake_binary && \
317
317
cmake \
318
- -DCMAKE_BUILD_TYPE=RELEASE \
318
+ -D BUILD_PERF_TESTS=OFF \
319
+ -D BUILD_TESTS=OFF \
319
320
-D BUILD_TIFF=ON \
320
321
-D BUILD_opencv_java=OFF \
322
+ -D CMAKE_BUILD_TYPE=RELEASE \
323
+ -D CMAKE_BUILD_TYPE=RELEASE \
324
+ -D CMAKE_INSTALL_PREFIX=$(python3 -c "import sys; print(sys.prefix)") \
321
325
-D CUDA_BIN_PATH=/usr/local/cuda \
322
326
-D CUDA_CUDA_LIBRARY=/usr/lib/x86_64-linux-gnu/stubs/libcuda.so \
323
327
-D CUDA_FAST_MATH=ON \
324
328
-D CUDA_HOST_COMPILER=/usr/bin/gcc-6 \
325
329
-D CUDA_NVCC_FLAGS="-D_FORCE_INLINES --expt-relaxed-constexpr" \
326
- -D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-10.1 \
327
- -D CUDNN_INCLUDE_DIR=/usr/local/cuda/include \
328
- -D CUDNN_LIBRARY=/usr/lib/x86_64-linux-gnu/libcudnn.so.8.0.5.39 \
329
- -D ENABLE_AVX=ON \
330
- -D WITH_CUDA=ON \
331
330
-D CUDA_NVCC_FLAGS=--expt-relaxed-constexpr \
332
331
-D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-11.3 \
332
+ -D CUDNN_INCLUDE_DIR=/usr/local/cuda/include \
333
+ -D CUDNN_LIBRARY=/usr/lib/x86_64-linux-gnu/libcudnn.so.8.2.0 \
334
+ -D ENABLE_AVX=ON \
333
335
-D ENABLE_FAST_MATH=1 \
334
- -D WITH_CUBLAS=1 \
335
- -D WITH_OPENGL=ON \
336
- -D WITH_OPENCL=ON \
337
- -D WITH_IPP=ON \
338
- -D WITH_TBB=ON \
339
- -D WITH_EIGEN=ON \
340
- -D WITH_V4L=ON \
341
- -D BUILD_TESTS=OFF \
342
- -D BUILD_PERF_TESTS=OFF \
343
336
-D OPENCV_EXTRA_MODULES_PATH="../../opencv_contrib-4.5.2/modules" \
344
- -D CMAKE_BUILD_TYPE=RELEASE \
345
- -D CMAKE_INSTALL_PREFIX=$(python3 -c "import sys; print(sys.prefix)") \
346
337
-D PYTHON_EXECUTABLE=$(which python3) \
347
338
-D PYTHON_INCLUDE_DIR=$(python3 -c "from distutils.sysconfig import get_python_inc; print(get_python_inc())") \
348
339
-D PYTHON_PACKAGES_PATH=$(python3 -c "from distutils.sysconfig import get_python_lib; print(get_python_lib())") \
340
+ -D WITH_CUBLAS=1 \
341
+ -D WITH_CUDA=ON \
342
+ -D WITH_EIGEN=ON \
343
+ -D WITH_IPP=ON \
344
+ -D WITH_OPENCL=ON \
345
+ -D WITH_OPENGL=ON \
346
+ -D WITH_TBB=ON \
347
+ -D WITH_V4L=ON \
349
348
.. 2>&1 | tee cmake_messages.txt && \
350
349
make -j$(nproc) && \
351
350
make install
0 commit comments