torch.cuda.synchronize — PyTorch 2.0 documentation (original) (raw)

• Docs >
• torch.cuda >
• torch.cuda.synchronize
• 

Shortcuts

torch.cuda.synchronize(device=None)[source]¶

Waits for all kernels in all streams on a CUDA device to complete.

Parameters:

device (torch.device or int, optional) – device for which to synchronize. It uses the current device, given by current_device(), if device is None (default).

• torch.cuda.synchronize

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.