1 parent 969be32 commit 37c2872Copy full SHA for 37c2872
1 file changed
whisperjav/utils/preflight_check.py
@@ -150,8 +150,9 @@ def _check_pytorch_cuda(self):
150
runtime_cuda = torch.version.cuda
151
152
# Convert compiled_cuda integer to version string for comparison
153
- # compiled_cuda is an integer like 12090 representing version 12.0.90
154
- # runtime_cuda is a string like "12.9"
+ # compiled_cuda is an integer like 12090 representing CUDA 12.9.0
+ # We extract major.minor (12.9) to match runtime_cuda format (string "12.9")
155
+ # Note: Patch version is intentionally truncated as runtime_cuda doesn't include it
156
if isinstance(compiled_cuda, int):
157
major = compiled_cuda // 1000
158
minor = (compiled_cuda % 1000) // 10
0 commit comments