Ищите любой пример Jetson Nano DockerHub, который использует графический процессор

Question

Я начинаю с моего Jetson Nano и ищу пример, который можно запустить, запустив docker run xxx, где xxx - это какой-то образ в DockerHub, использующий графический процессор.

Я предполагаю, что яПридется передать некоторые флаги --device, но есть ли какой-нибудь готовый образец стиля "hello world", использующий графический процессор из докера?

Я надеюсь просто продемонстрировать, чтоВы можете получить доступ к графическому процессору из док-контейнера на Jetson Nano.Главным образом, чтобы убедиться, что моя конфигурация правильная.

Answer 1

Вы не одиноки, желая этого, но вы не можете сделать это в данный момент.Команда NVIDIA Nano осознает необходимость, и эта функция ожидается позже в этом году.

См. https://devtalk.nvidia.com/default/topic/1050809/jetson-nano/docker-image-to-see-if-cuda-is-working-in-container-on-jetson-nano-/

В настоящее время вы можете запустить докер-контейнер с установленными TensorFlow или PyTorch.но он будет использовать только процессор, а не графический процессор.