TypeError: _queue_reduction(): incompatible function arguments.

使用pytorch的DDP分布式训练的时候遇到错误： TypeError: _queue_reduction(): incompatible function arguments. The following argument types are supported: 1. (process_group: torch.distributed.ProcessGroup, grads_batch: List[List[at::Tensor]], devices: List[int]) -> Tuple[torch.distributed.Work, at::Tensor] Invoked