PyTorch:NVIDIA NGC映像还是Docker Hub映像?

5
1个回答

2

我不是 docker 专家。
据我所知,Nvidia 做了很多努力来提供 GPU 优化的容器,因此在 Nvidia GPU 上使用 Nvidia 容器运行 GPU PyTorch 应该有最佳性能。
因此,如果您正在使用 Nvidia 硬件,则应该希望使用 NGC 容器可以获得更好的性能。然而,差距可能不是那么显著。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接