Как включить ускорение 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?>

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