PyTorch: NVIDIA NGC镜像还是Docker Hub镜像？

Question

the official PyTorch image on Docker Hub和the PyTorch image on NVIDIA NGC之间的区别是什么

NGC页面比Docker Hub页面更具文档化，后者没有说明。但是NGC镜像也要重几to，而且它似乎需要一个与CUDA 10.2兼容的驱动程序。

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

Answer 1

我不是码头专家。

据我所知，Nvidia花了很多精力来发布GPU优化的容器，以便在带有Nvidia容器的Nvidia GPU上运行GPU pytorch应该具有最好的性能。

因此，如果您使用的是Nvidia硬件，那么使用NGC容器会获得更好的性能。然而，差距可能并不是那么大。