Как обеспечить ускорение MPS для Pytorch внутри Docker на Mac? ⇐ Python

[Anonymous]

Я пытаюсь настроить Docker для проекта Python на моем Mac и хочу использовать MPS (металлические шейды) для ускорения GPU с помощью Pytorch внутри контейнера. < /p>
То, что я пробовал до сих пор: < /p>

Я установил Docker Dayptop (последняя версия Apple Silon). /> Создал Dockerfile, используя базовое изображение, такое как Python: 3.11-SLIM. < /li>
Установлен Pytorch внутри контейнера и проверен на наличие MPS
с: < /li>
< /ul>
import torch
print(torch.backends.mps.is_available())
< /code>
Это возвращает false < /code>, означает, что MPS не обнаруживается внутри контейнера. < /p>

Запуск одного и того же кода вне докера на macOS (голый металл), MPS доступен (TRUE). Docker on MacOS Поддержка GPU GPU для ускорения MPS? Docker? Спасибо!

Подробнее здесь: https://stackoverflow.com/questions/795 ... ker-on-mac