Remove outdated CUDA version check (#148142)

Since Torch requires CUDA>=11, some checks can be removed.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/148142
Approved by: https://github.com/janeyx99, https://github.com/eqy
This commit is contained in:
cyy
2025-03-04 03:33:39 +00:00
committed by PyTorch MergeBot
parent 98bf2f1170
commit ec2805ada8
6 changed files with 15 additions and 30 deletions

View File

@ -2226,7 +2226,6 @@ class DistributedDataParallel(Module, Joinable):
if hook.__name__ in ["bf16_compress_hook", "bf16_compress_wrapper_hook"]:
cuda_supported = (
torch.version.cuda is not None
and int(torch.version.cuda.split(".")[0]) >= 11
) or torch.version.hip is not None
nccl_supported = (
dist.is_available()