torch.cuda.ipc_collect

torch.cuda.ipc_collect()[源代码]

在CUDA IPC释放后，强制收集GPU内存。

注意

检查是否可以将任何已发送的CUDA张量从内存中清除。如果没有活动的计数器，会强制关闭用于引用计数的共享内存文件。这对于生产进程在停止主动发送张量后释放未使用的内存非常有用。