torch.xpu.memory.mem_get_info#
- torch.xpu.memory.mem_get_info(device=None)[原始碼]#
返回給定裝置的全域性可用和總 GPU 記憶體。
- 引數
device (torch.device 或 int 或 str, 可選) – 選定的裝置。如果
device為None（預設值），則返回當前裝置（由current_device()提供）的統計資訊。- 返回
裝置上的可用記憶體（位元組為單位）。 int: 裝置上的總記憶體（位元組為單位）
- 返回型別
