Respect current GPU minor version

2025-11-01 13:34:57 +08:00 · 2025-10-06 10:30:07 +09:00
1 changed files with 1 additions and 1 deletions
--- a/torch/cuda/init.py
+++ b/torch/cuda/init.py
@ -280,7 +280,7 @@ def _check_capability():
                (_extract_arch_version(arch) for arch in torch.cuda.get_arch_list()),
                default=50,
            )
-            if current_arch < min_arch or current_arch > max_arch:
+            if current_arch + minor < min_arch or current_arch > max_arch:
                warnings.warn(
                    incompatible_gpu_warn
                    % (