【发布时间】: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 的映像有什么优势吗?
【问题讨论】:
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 的映像有什么优势吗?
【问题讨论】:
我不是码头工人专家。
据我所知,Nvidia 付出了很多努力来发布 GPU 优化的容器,以便在具有 Nvidia 容器的 Nvidia GPU 上运行 GPU pytorch 应该具有最佳性能。
因此,如果您使用的是 Nvidia 硬件,您应该期望使用 NGC 容器获得更好的性能。不过,差距可能没有那么大。
【讨论】: