From 05a66dcac73c84211aec4cfe7a4e3aef9670c115 Mon Sep 17 00:00:00 2001 From: deadprogram Date: Mon, 9 Oct 2023 15:34:26 +0200 Subject: [PATCH] docker: exclude deprecated cuda architectures Signed-off-by: deadprogram --- Dockerfile.opencv-gpu-cuda-11 | 2 +- Dockerfile.opencv-gpu-cuda-11.2.2 | 2 +- Dockerfile.opencv-gpu-cuda-12 | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/Dockerfile.opencv-gpu-cuda-11 b/Dockerfile.opencv-gpu-cuda-11 index 89bb8a51..15d5506a 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_GENERATION=Auto \ + -D CUDA_ARCH_BIN=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-11.2.2 b/Dockerfile.opencv-gpu-cuda-11.2.2 index 61733561..e0f6dc3a 100644 --- a/Dockerfile.opencv-gpu-cuda-11.2.2 +++ b/Dockerfile.opencv-gpu-cuda-11.2.2 @@ -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_GENERATION=Auto \ + -D CUDA_ARCH_BIN=5.2,6.0,6.1,7.0,7.5,8.0,8.6 \ -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 619b503f..18fc73d9 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_GENERATION=Auto \ + -D CUDA_ARCH_BIN=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 && \