Wie aktiviere ich die MPS-Beschleunigung für PyTorch in Docker auf dem Mac? ⇐ Python

[Anonymous]

Ich versuche, Docker für ein Python-Projekt auf meinem Mac einzurichten und möchte MPS (Metal Performance Shaders) für die GPU-Beschleunigung mit PyTorch im Container verwenden.
Was ich bisher versucht habe:

• Ich habe Docker Desktop (die neueste Version von Apple Silicon) unter MacOS installiert.
• Erstellte eine Docker-Datei mit einem Basis-Image wie python:3.11-slim.
• PyTorch im Container installiert und auf MPS
  Verfügbarkeit geprüft mit:

Code: Select all

import torch
print(torch.backends.mps.is_available())

Dies gibt False zurück, was bedeutet, dass MPS nicht im Container erkannt wird.

• Wenn derselbe Code außerhalb von Docker unter macOS (Bare Metal) ausgeführt wird, ist MPS verfügbar (True).

Unterstützt Docker unter macOS GPU-Passthrough für MPS-Beschleunigung?