Test with PyTorch 22.02 docker

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