diff --git a/Dockerfile.opencv-gpu-cuda-11 b/Dockerfile.opencv-gpu-cuda-11 index 54d030ff..5cd1d1a0 100644 --- a/Dockerfile.opencv-gpu-cuda-11 +++ b/Dockerfile.opencv-gpu-cuda-11 @@ -57,8 +57,9 @@ RUN curl -Lo opencv.zip ${OPENCV_FILE} && \ -D WITH_CUDNN=ON \ -D OPENCV_DNN_CUDA=ON \ -D CUDA_GENERATION=Auto \ + -D CUDA_ARCH_BIN=5.3,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 --ignore 2) && \ + make -j $(nproc --all) && \ make preinstall && make install && ldconfig && \ cd / && rm -rf opencv* diff --git a/Dockerfile.opencv-gpu-cuda-11.2.2 b/Dockerfile.opencv-gpu-cuda-11.2.2 index 61733561..19d3f260 100644 --- a/Dockerfile.opencv-gpu-cuda-11.2.2 +++ b/Dockerfile.opencv-gpu-cuda-11.2.2 @@ -57,6 +57,7 @@ RUN curl -Lo opencv.zip ${OPENCV_FILE} && \ -D WITH_CUDNN=ON \ -D OPENCV_DNN_CUDA=ON \ -D CUDA_GENERATION=Auto \ + -D CUDA_ARCH_BIN=5.3,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 619b503f..fd041e45 100644 --- a/Dockerfile.opencv-gpu-cuda-12 +++ b/Dockerfile.opencv-gpu-cuda-12 @@ -57,6 +57,7 @@ RUN curl -Lo opencv.zip ${OPENCV_FILE} && \ -D WITH_CUDNN=ON \ -D OPENCV_DNN_CUDA=ON \ -D CUDA_GENERATION=Auto \ + -D CUDA_ARCH_BIN=5.3,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 && \