Skip to content

Commit e2449ee

Browse files
committed
fixed cudnn version
1 parent 3ca2a26 commit e2449ee

File tree

1 file changed

+19
-20
lines changed

1 file changed

+19
-20
lines changed

commons/Dockerfile.base.py38-cuda11.3

Lines changed: 19 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -129,9 +129,9 @@ RUN rm -fr /var/lib/apt/lists/* && \
129129
ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 && \
130130
echo "/usr/local/cuda/lib64/stubs" > /etc/ld.so.conf.d/z-cuda-stubs.conf && \
131131

132-
# Install CUDA-11.3 + cuDNN 8.0.5
132+
# Install CUDA-11.3 + cuDNN 8.2.0
133133
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 && \
135135
ldconfig
136136

137137
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 && \
278278
rm -f /tmp/*.whl /tmp/requirements.py38.txt
279279

280280
# 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 && \
282282
apt-get install -y \
283283
libgstreamer1.0-dev \
284284
libgstreamer-plugins-base1.0-dev \
@@ -315,37 +315,36 @@ RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
315315
mkdir opencv-${OPENCV_VERSION}/cmake_binary && \
316316
cd opencv-${OPENCV_VERSION}/cmake_binary && \
317317
cmake \
318-
-DCMAKE_BUILD_TYPE=RELEASE \
318+
-D BUILD_PERF_TESTS=OFF \
319+
-D BUILD_TESTS=OFF \
319320
-D BUILD_TIFF=ON \
320321
-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)") \
321325
-D CUDA_BIN_PATH=/usr/local/cuda \
322326
-D CUDA_CUDA_LIBRARY=/usr/lib/x86_64-linux-gnu/stubs/libcuda.so \
323327
-D CUDA_FAST_MATH=ON \
324328
-D CUDA_HOST_COMPILER=/usr/bin/gcc-6 \
325329
-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 \
331330
-D CUDA_NVCC_FLAGS=--expt-relaxed-constexpr \
332331
-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 \
333335
-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 \
343336
-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)") \
346337
-D PYTHON_EXECUTABLE=$(which python3) \
347338
-D PYTHON_INCLUDE_DIR=$(python3 -c "from distutils.sysconfig import get_python_inc; print(get_python_inc())") \
348339
-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 \
349348
.. 2>&1 | tee cmake_messages.txt && \
350349
make -j$(nproc) && \
351350
make install

0 commit comments

Comments
 (0)