目录表

快捷键

torch.xpu.memory_reserved¶

torch.xpu.memory_reserved(device=None)[source][source]¶

返回给定设备由缓存分配器管理的当前 GPU 内存大小（以字节为单位）。

参数:: device (torch.device 或 int 或 str，可选) – 选择设备。如果 device 是 None （默认），则返回当前设备的统计信息，由 current_device() 指定。
返回类型:: int

文档

PyTorch 的全面开发者文档

查看文档

教程

深入了解初学者和高级开发者的教程

查看教程

资源

查找开发资源并获得您的疑问解答

查看资源