docker: specfic more limited set to build for cuda 12

Signed-off-by: deadprogram <[email protected]>
hybridgroup · Oct 10, 2023 · 48f4c53 · 48f4c53
1 parent f8ec0cf
commit 48f4c53
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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=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 && \