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



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

我想我将不得不传入一些 --device 标志,但是否有任何一种"hello world"风格的示例可以使用 docker 的 GPU?

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

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托管映像:

你并不是唯一一个想要这样做的人,但你目前还做不到。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。

最新更新