Is your feature request related to a problem? Please describe.
Thanks for @deepib 's reminder, PyTorch 22.02 docker is available now:
https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch
I will try to update the CI tests config and run all unit tests & integration tests to verify it.

Thanks.