2

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

我假设我必须传入一些 --device 标志，但是是否有任何类型的“hello world”风格的样本准备好使用来自 docker 的 GPU？

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

4

2 回答 2

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。

于 2019-05-13T08:00:12.243 回答
1

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

于 2019-09-26T09:30:36.887 回答