docker - PyTorch：NVIDIA NGC 映像还是 Docker Hub 映像？

Question

Docker Hub 上的官方 PyTorch 镜像和NVIDIA NGC 上的 PyTorch 镜像有什么区别？

NGC 页面比没有描述的 Docker Hub 页面记录更多。但 NGC 映像也重了几 GB，而且似乎需要兼容 CUDA 10.2 的驱动程序。

使用 NVIDIA NGC 图像而不是来自 Docker Hub 的图像有什么优势吗？

标签： dockerpytorchnvidia

Answer 1

我不是码头专家。
据我所知，Nvidia 付出了很多努力来发布 GPU 优化的容器，以便在具有 Nvidia 容器的 Nvidia GPU 上运行 GPU pytorch 应该具有最佳性能。
因此，如果您使用的是 Nvidia 硬件，您应该期望使用 NGC 容器获得更好的性能。不过，差距可能并不那么大。