Update CUDA archs to 5.0;6.0;7.0;8.0;9.0 in builds for OpenCV, ONNX R…

…untime, PyTorch
bytedeco · Nov 14, 2023 · cfcc2d5 · cfcc2d5
1 parent 38ac695
commit cfcc2d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/opencv/cppbuild.sh b/opencv/cppbuild.sh
@@ -133,7 +133,7 @@ BUILD_CONTRIB_X="-DBUILD_opencv_stereo=OFF -DBUILD_opencv_plot=ON -DBUILD_opencv
 
 GPU_FLAGS="-DWITH_CUDA=OFF"
 if [[ "$EXTENSION" == *gpu ]]; then
-    GPU_FLAGS="-DWITH_CUDA=ON -DWITH_CUDNN=ON -DOPENCV_DNN_CUDA=ON -DCUDA_VERSION=12.3 -DCUDNN_VERSION=8.9 -DCUDA_ARCH_BIN='5.0;6.0;7.0' -DCUDA_ARCH_PTX='7.0' -DCUDA_NVCC_FLAGS=--expt-relaxed-constexpr -DCUDA_nppicom_LIBRARY="
+    GPU_FLAGS="-DWITH_CUDA=ON -DWITH_CUDNN=ON -DOPENCV_DNN_CUDA=ON -DCUDA_VERSION=12.3 -DCUDNN_VERSION=8.9 -DCUDA_ARCH_BIN='5.0;6.0;7.0;8.0;9.0' -DCUDA_ARCH_PTX='' -DCUDA_NVCC_FLAGS=--expt-relaxed-constexpr -DCUDA_nppicom_LIBRARY="
 fi
 
 # exclude openblas dependencies