torch.mps¶
此包提供了一个接口,用于在 Python 中访问 MPS(Metal Performance Shaders)后端。Metal 是 Apple 用于编程金属 GPU(图形处理单元)的 API。使用 MPS 意味着可以通过在金属 GPU 上运行工作来提高性能。有关更多详细信息,请参阅 https://developer.apple.com/documentation/metalperformanceshaders。
返回可用 MPS 设备的数量。 |
|
等待 MPS 设备上所有流中的所有内核完成。 |
|
将随机数生成器状态作为 ByteTensor 返回。 |
|
设置随机数生成器状态。 |
|
设置生成随机数的种子。 |
|
将生成随机数的种子设置为随机数。 |
|
释放缓存分配器当前持有的所有未使用的缓存内存,以便这些内存可以在其他 GPU 应用程序中使用。 |
|
设置内存分数以限制进程在 MPS 设备上的内存分配。 |
|
以字节为单位返回张量当前占用的 GPU 内存。 |
|
以字节为单位返回 Metal 驱动程序为进程分配的总 GPU 内存。 |
|
以字节为单位返回建议的 GPU 内存最大工作集大小。 |
MPS 分析器¶
从 MPS 后端开始 OS Signpost 跟踪。 |
|
停止从 MPS 后端生成 OS Signpost 跟踪。 |
|
上下文管理器,用于启用从 MPS 后端生成 OS Signpost 跟踪。 |
MPS 事件¶
MPS 事件的包装器。 |