mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
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:
@ -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()
|
||||
|
Reference in New Issue
Block a user