torch.cuda.ipc_collect

torch.cuda.ipc_collect() [source]

Принудительно собирает память GPU после того, как она была освобождена с помощью CUDA IPC.

Примечание

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