torch.cuda.synchronize

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

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

Параметры:

device (torch.device или int, необязательно) – устройство для синхронизации. Использует текущее устройство, заданное current_device(), если device равно None (по умолчанию).