Как сделать образ docker с помощью Anaconda и использовать GPU?

Question

У меня есть графический процессор и драйвер CUDA. Я создаю docker изображение, используя официальное изображение anaconda, доступное в docker hub. Когда я использую команду nvidia-smi внутри образа docker, он показывает, что не установлен драйвер CUDA.

Как создать образ docker, чтобы использовать Anaconda с графическим процессором?

Answer 1

Docker 19.03 имеет встроенную поддержку графических процессоров NVIDIA. Установите nvidia-container-runtime и используйте флаг docker run --gpus all с вашим контейнером Anaconda. Помните, что вы можете запускать Windows контейнеры с ускорением GPU на хосте Windows, используя Docker 19.03, но не контейнер Linux.

Источник: Ускорение GPU в Windows Контейнеры