torch.cuda.ipc_collect#

torch.cuda.ipc_collect()[source]#

Force collects GPU memory after it has been released by CUDA IPC.

Note

Checks if any sent CUDA tensors could be cleaned from the memory. Forcecloses shared memory file used for reference counting if there is noactive counters. Useful when the producer process stopped actively sendingtensors and want to release unused memory.