would be great to use a more recent base image than the current 22.10:
https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch/tags

once this ticket is done, #5777 should be verified and closed as that's an issue with pytorch 22.10.

cc @Nic-Ma @YanxuanLiu