Skip to content

Commit 6554339

Browse files
committed
fixed cudnn path for opencv
1 parent 4f4d936 commit 6554339

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

commons/Dockerfile.base.py38-cuda11.3

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -558,7 +558,7 @@ RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
558558
-D CUDA_NVCC_FLAGS="-D_FORCE_INLINES --expt-relaxed-constexpr" \
559559
-D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-11.3 \
560560
-D CUDNN_INCLUDE_DIR=/usr/local/cuda/include \
561-
-D CUDNN_LIBRARY=/usr/lib/x86_64-linux-gnu/libcudnn.so \
561+
-D CUDNN_LIBRARY=/usr/lib/x86_64-linux-gnu/libcudnn.so.8.2.0 \
562562
-D ENABLE_AVX=ON \
563563
-D ENABLE_FAST_MATH=1 \
564564
-D HAVE_opencv_python3=ON \

0 commit comments

Comments
 (0)