**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.
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.