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

Question

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

NGC 页面比 Docker Hub 页面有更多文档，后者没有任何描述。但 NGC 图像也重了几 GB，并且似乎需要 CUDA 10.2 兼容的驱动程序。

使用 NVIDIA NGC 镜像代替 Docker Hub 镜像有什么优势吗？

Answer 1

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