torch.mps

此包启用了用于在 python 中访问 MPS 后端的接口

同步	等待 MPS 设备上所有流中的所有内核完成。
get_rng_state	将随机数生成器状态作为 ByteTensor 返回。
set_rng_state	设置随机数生成器状态。
manual_seed	设置用于生成随机数的种子。
种子	将用于生成随机数的种子设置为随机数。
empty_cache	释放缓存分配器当前持有的所有未占用的缓存内存，以便这些内存可以在其他 GPU 应用程序中使用。
set_per_process_memory_fraction	设置内存分数以限制进程在 MPS 设备上的内存分配。
current_allocated_memory	返回张量占用的当前 GPU 内存（以字节为单位）。
driver_allocated_memory	返回 Metal 驱动程序为进程分配的总 GPU 内存（以字节为单位）。