From e21c5c4ca90b613bc8ea6e4f2cfbaf11313e4671 Mon Sep 17 00:00:00 2001 From: deadprogram Date: Mon, 9 Oct 2023 20:09:53 +0200 Subject: [PATCH] docker: avoid deprecated cuda architectures Signed-off-by: deadprogram --- Dockerfile.opencv-gpu-cuda-11 | 2 +- Dockerfile.opencv-gpu-cuda-12 | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile.opencv-gpu-cuda-11 b/Dockerfile.opencv-gpu-cuda-11 index bc0108c2..68deb737 100644 --- a/Dockerfile.opencv-gpu-cuda-11 +++ b/Dockerfile.opencv-gpu-cuda-11 @@ -56,7 +56,7 @@ RUN curl -Lo opencv.zip ${OPENCV_FILE} && \ -D BUILD_opencv_cudacodec=OFF \ -D WITH_CUDNN=ON \ -D OPENCV_DNN_CUDA=ON \ - -D CUDA_ARCH_BIN=5.3,6.0,6.1,7.0,7.5,8.0,8.6,8.9,9.0 \ + -D CUDA_ARCH_BIN=5.0,5.2,6.0,6.1,7.0,7.5,8.0,8.6,8.9,9.0 \ -D OPENCV_GENERATE_PKGCONFIG=ON .. && \ make -j $(nproc --all) && \ make preinstall && make install && ldconfig && \ diff --git a/Dockerfile.opencv-gpu-cuda-12 b/Dockerfile.opencv-gpu-cuda-12 index 18533c1a..619b503f 100644 --- a/Dockerfile.opencv-gpu-cuda-12 +++ b/Dockerfile.opencv-gpu-cuda-12 @@ -56,7 +56,7 @@ RUN curl -Lo opencv.zip ${OPENCV_FILE} && \ -D BUILD_opencv_cudacodec=OFF \ -D WITH_CUDNN=ON \ -D OPENCV_DNN_CUDA=ON \ - -D CUDA_ARCH_BIN=5.3,6.0,6.1,7.0,7.5,8.0,8.6,8.9,9.0 \ + -D CUDA_GENERATION=Auto \ -D OPENCV_GENERATE_PKGCONFIG=ON .. && \ make -j $(nproc --all) && \ make preinstall && make install && ldconfig && \