torch.mps

這個套件提供了一個介面，可以使用 Python 存取 MPS（Metal Performance Shaders）後端。Metal 是 Apple 的 API，用於編寫金屬 GPU（圖形處理單元）程式。使用 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 記憶體，以位元組為單位。

MPS 分析器

profiler.start	從 MPS 後端開始作業系統 Signpost 追蹤。
profiler.stop	停止從 MPS 後端產生作業系統 Signpost 追蹤。
profiler.profile	上下文管理器，用於啟用從 MPS 後端產生作業系統 Signpost 追蹤。

MPS 事件

event.Event

MPS 事件的包裝器。