Is your feature request related to a problem? Please describe.
Deepi asked about the support for new PyTorch docker 22.01:
https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch/tags
I will try to submit a PR and run the full tests.