我正在开始使用我的 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。