Is your feature request related to a problem? Please describe.
As the new PyTorch docker had been released: https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch.
It would be nice to update our dependency and CI tests to cover it.