Skip to content

Commit 86786e3

Browse files
committed
[dist] Bump minimum PTX to 7.5 to control build times
We can consider creating two distributions as well.
1 parent 021ab91 commit 86786e3

File tree

3 files changed

+7
-4
lines changed

3 files changed

+7
-4
lines changed

.claude/settings.local.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@
55
"permissions": {
66
"allow": [
77
"Bash(gh run view:*)",
8-
"Bash(gh workflow:*)"
8+
"Bash(gh workflow:*)",
9+
"Bash(unzip:*)"
910
]
1011
}
1112
}

.github/workflows/build_wheels_windows.yml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,8 @@ jobs:
4343
cuda-version: ['12']
4444
include:
4545
# CUDA 12: Supports Maxwell (5.0) through Blackwell (10.0)
46+
# TODO(@Breakthrough): Expand PTX legacy support back to 5.0, it doesn't make the final
47+
# distribution that much larger and may help users with older GPUs.
4648
- cuda-version: '12'
4749
runs-on: 'windows-2025'
4850
cuda-installer: 'cuda_12.9.1_windows_network.exe'
@@ -51,7 +53,7 @@ jobs:
5153
cudnn-folder: 'cudnn-windows-x86_64-9.18.1.3_cuda12-archive'
5254
video-codec-sdk-archive: 'Video_Codec_SDK_13.0.37.zip'
5355
video-codec-sdk-folder: 'Video_Codec_SDK_13.0.37'
54-
cuda-arch-bin: '5.0;5.2;6.0;6.1;7.0;7.5;8.0;8.6;8.9;9.0;10.0'
56+
cuda-arch-bin: '7.5;8.0;8.6;8.9;9.0;10.0'
5557
cuda-arch-ptx: '10.0'
5658
cache-key: 'nvidia-deps-cuda-12.9.1-cudnn-9.18.1.3'
5759
# TODO: Re-enable CUDA 13 after updating to OpenCV 4.13+
@@ -137,7 +139,7 @@ jobs:
137139
$CUDNN_PATH = "D:/a/opencv-python-cuda/opencv-python-cuda/${{ matrix.cudnn-folder }}"
138140
echo "CUDNN_LIBRARY=$CUDNN_PATH/lib/x64/cudnn.lib" | Out-File -FilePath $env:GITHUB_ENV -Append
139141
echo "CUDNN_INCLUDE_DIR=$CUDNN_PATH/include" | Out-File -FilePath $env:GITHUB_ENV -Append
140-
Copy-Item -Path "$CUDNN_PATH/bin/*" -Destination . -Include "*.dll"
142+
Copy-Item -Path "$CUDNN_PATH/bin/x64/*" -Destination . -Include "*.dll"
141143
shell: pwsh
142144
- name: 🔧 Install NVIDIA Video Codec SDK
143145
run: |

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ Download: [Latest Release (4.12.0-dev0)](https://github.com/Breakthrough/opencv-
44

55
## What is `opencv-python-cuda`?
66

7-
Pre-built NVIDIA® CUDA™ enabled OpenCV packages for Python that come with all batteries included. This is a fork of [the official opencv-python project](https://github.com/opencv/opencv-python). Right now packages are only produced for Windows x64, and devices must be Maxwell class (GeForce 900 series) or newer. Once installed via `pip` (or another Python package manager like `uv`), the following should *just work*:
7+
Pre-built NVIDIA® CUDA™ enabled OpenCV packages for Python that come with all batteries included. This is a fork of [the official opencv-python project](https://github.com/opencv/opencv-python). Right now packages are only produced for Windows x64, and devices must be Turing class (GeForce 2000 series) or newer. Once installed via `pip` (or another Python package manager like `uv`), the following should *just work*:
88

99
```
1010
import cv2

0 commit comments

Comments
 (0)