torch.cuda.synchronize

torch.cuda.synchronize(device=None)

等待 CUDA 裝置上所有串流中的所有核心完成。

參數

device (torch.device 或 int, optional) – 要同步的裝置。 如果 device 為 None (預設)，它會使用 current_device() 給定的當前裝置。