目錄

快捷方式

torch.cuda.get_per_process_memory_fraction¶

torch.cuda.get_per_process_memory_fraction(device=None)[來源][來源]¶

取得程序的記憶體比例。

參數: device (torch.device 或 int, 選用) – 選取的裝置。如果它是 None，則會使用預設的 CUDA 裝置。
傳回: 記憶體比例，範圍為 0~1。允許的記憶體等於 total_memory * fraction。
傳回類型: float

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學

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

檢視教學課程

資源

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

檢視資源