torch.cuda.ipc_collect

torch.cuda.ipc_collect() [исходный код]

Принудительно очищает память GPU после её освобождения посредством CUDA IPC.

Примечание

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