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

[Anonymous]

Я пытаюсь настроить Docker для проекта Python на своем Mac и хочу использовать MPS (Metal Performance Shaders) для ускорения графического процессора с помощью PyTorch внутри контейнера.
Что я пробовал до сих пор:

• Я установил Docker Desktop (последнюю версию Apple Silicon) на MacOS.
• Создал Dockerfile, используя базовый образ, например python:3.11-slim.
• Установил PyTorch внутри контейнера и проверил доступность MPS
  с помощью:

Код: Выделить всё

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

Это возвращает False, что означает, что MPS не обнаружен внутри контейнера.

• При запуске того же кода вне Docker на macOS (голое железо), MPS доступен (True).

Мои вопросы:

• Поддерживает ли Docker в macOS сквозную передачу графического процессора для MPS ускорение?
• Существуют ли какие-либо конкретные настройки или конфигурации Docker для включения доступа к графическому процессору внутри контейнера?
• Если сквозная передача графического процессора не поддерживается, существуют ли какие-либо обходные пути для использования MPS с PyTorch внутри Docker?

Будем очень признательны за любые рекомендации или обходные пути. Спасибо!

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