快捷方式

torch.mps

此包提供了一个接口,用于在 Python 中访问 MPS(Metal Performance Shaders)后端。Metal 是 Apple 用于编程金属 GPU(图形处理单元)的 API。使用 MPS 意味着可以通过在金属 GPU 上运行工作来提高性能。有关更多详细信息,请参阅 https://developer.apple.com/documentation/metalperformanceshaders

device_count

返回可用 MPS 设备的数量。

synchronize

等待 MPS 设备上所有流中的所有内核完成。

get_rng_state

将随机数生成器状态作为 ByteTensor 返回。

set_rng_state

设置随机数生成器状态。

manual_seed

设置生成随机数的种子。

seed

将生成随机数的种子设置为随机数。

empty_cache

释放缓存分配器当前持有的所有未使用的缓存内存,以便这些内存可以在其他 GPU 应用程序中使用。

set_per_process_memory_fraction

设置内存分数以限制进程在 MPS 设备上的内存分配。

current_allocated_memory

以字节为单位返回张量当前占用的 GPU 内存。

driver_allocated_memory

以字节为单位返回 Metal 驱动程序为进程分配的总 GPU 内存。

recommended_max_memory

以字节为单位返回建议的 GPU 内存最大工作集大小。

MPS 分析器

profiler.start

从 MPS 后端开始 OS Signpost 跟踪。

profiler.stop

停止从 MPS 后端生成 OS Signpost 跟踪。

profiler.profile

上下文管理器,用于启用从 MPS 后端生成 OS Signpost 跟踪。

MPS 事件

event.Event

MPS 事件的包装器。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

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

查看资源