寻找任何使用GPU的Jetson Nano DockerHub示例

Question

我正在开始使用Jetson Nano，我正在寻找一个可以运行docker run xxx来启动的示例，其中xxx是DockerHub上使用GPU的某些图像。

我假设我必须传递一些--device标志，但是甚至还有准备使用docker中的GPU的任何类型的“ hello world”风格示例吗？

我希望证明您可以从Jetson Nano上的docker容器访问GPU。主要是为了确保我的配置正确。

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的Docker容器。但是它将仅使用CPU，而不使用GPU。

Answer 2

Nvidia Jetpack 4.2.1在Jetson Nano上启用了带有GPU的简单Docker。

有关如何使Docker和Kubernetes在具有GPU的Jetson Nano上运行的详细说明，请参见此处：

https://medium.com/jit-team/building-a-gpu-enabled-kubernets-cluster-for-machine-learning-with-nvidia-jetson-nano-7b67de74172a

它为TensorFlow使用了一个简单的Docker Hub托管映像：