PyTorch: NVIDIA NGC image or Docker Hub image?

user_5248987 0
Viewed 599

What are the differences between the official PyTorch image on Docker Hub and the PyTorch image on NVIDIA NGC?

The NGC page is more documented than the Docker Hub page, which has no description. But the NGC image is also heavier by a few gigabytes, and it seems to require a CUDA 10.2-compatible driver.

Is there any advantage in using the NVIDIA NGC image instead of the one from Docker Hub?

pytorch docker nvidia
1 Answers

I'm not a docker expert.
To the best of my knowledge Nvidia puts a lot of effort to ship GPU-optimized containers such that running GPU pytorch on Nvidia GPU with Nvidia container should have best possible performance.
Therefore, if you are using Nvidia hardware you should expect better performance using NGC containers. The gap, though, might not be that significant.

user_17144100
Related