torch.cuda.ipc_collect

torch.cuda.ipc_collect()

強制回收已被 CUDA IPC 釋放的 GPU 記憶體。

注意

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