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

Posted 2023-04-15

【中文标题】PyTorch：NVIDIA NGC 映像还是 Docker Hub 映像？

【英文标题】：PyTorch: NVIDIA NGC image or Docker Hub image?

【发布时间】：2020-11-08 17:03:07

【问题描述】：

the official PyTorch image on Docker Hub 和the PyTorch image on NVIDIA NGC 有什么区别？

NGC 页面比 Docker Hub 页面记录更多，后者没有描述。但是NGC镜像也重了几G，而且好像需要CUDA 10.2兼容的驱动。

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

【参考方案1】：

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