torch.cuda.synchronize

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

Дождаться завершения всех ядер во всех потоках на устройстве CUDA.

Параметры:

device (torch.device or int, optional) – устройство, для которого выполняется синхронизация. Если device — это None (по умолчанию), используется текущее устройство, определяемое с помощью current_device().