Hi,
It’s recommended to use l4t-pytorch and l4t-ml containers from NGC for pyTorch v1.6.
l4t-pytorch
l4t-ml
For more detail, please check this topic:
Thanks.