捷徑

torch.mps.current_allocated_memory

torch.mps.current_allocated_memory()[原始碼][原始碼]

傳回目前 Tensor 佔用的 GPU 記憶體,以位元組為單位。

注意

傳回的大小不包括 MPSAllocator 記憶體池中快取的配置。

傳回類型

int

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學

取得針對初學者和進階開發者的深入教學

檢視教學

資源

尋找開發資源並獲得您問題的解答

檢視資源