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

Looking for any Jetson Nano DockerHub example that uses the GPU

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

我想我必须传递一些 --device 标志,但是否有任何类型的 "hello world" 风格的示例准备好使用来自 docker 的 GPU?

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

你不是唯一想要那个的人,但你现在做不到。 NVIDIA Nano 团队意识到了这一需求,预计该功能将在今年晚些时候推出。

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

目前您可以运行一个docker安装有TensorFlow或PyTorch的容器。但它只会使用 CPU,不使用 GPU。

Nvidia Jetpack 4.2.1 在 Jetson Nano 上使用 GPU 轻松 Docker。

有关如何使用 GPU 在 Jetson Nano 上获取 Docker 和 Kubernetes 运行 的详细说明,请参见此处:

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

它为 TensorFlow 使用一个简单的 Docker Hub 托管图像: