torch.cuda.ipc_collect

torch.cuda.ipc_collect() [source]

Вынужденное освобождение памяти GPU после её освобождения с помощью CUDA IPC.

Примечание

Проверяет, можно ли очистить из памяти любые отправленные тензоры CUDA. Вынужденно закрывает общий файл памяти, используемый для подсчёта ссылок, если активных счётчиков нет. Полезно, когда процесс-производитель перестал активно отправлять тензоры и хочет освободить неиспользуемую память.