torch.cuda.ipc_collect

torch.cuda.ipc_collect()

在 CUDA IPC 釋放 GPU 記憶體後強制收集。

注意

檢查是否有已傳送的 CUDA 張量可以從記憶體中清理。如果不存在活動的計數器，則強制關閉用於引用計數的共享記憶體檔案。當生產者程序停止主動傳送張量並希望釋放未使用的記憶體時，此功能非常有用。