docker: exclude deprecated cuda architectures

Signed-off-by: deadprogram <[email protected]>
hybridgroup · Oct 9, 2023 · 6a8954f · 6a8954f
1 parent a779e38
commit 6a8954f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git 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
@@ -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
@@ -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 && \