Unable to run yolox-pose in jetson orin AGX

[tet28]

Hi.
I'm trying to operate yolox-pose in jetson orin AGX but error raised.

~/work/test/testdetection/mmpose-main/projects/yolox-pose$ python demo/inferencer_demo.py ../../tests/data/coco/000000000785.jpg --pose2d configs/yolox-pose_s_8xb32-300e_coco.py --pose2d-weights https://download.openmmlab.com/mmpose/v1/projects/yolox-pose/yolox-pose_s_8xb32-300e_coco-9f5e3924_20230321.pth --scope mmyolo --vis-out-dir vis_results
Traceback (most recent call last):
File "demo/inferencer_demo.py", line 5, in
from mmpose.apis.inferencers import MMPoseInferencer, get_model_aliases
File "/home/user/work/test/testdetection/test-env/lib/python3.8/site-packages/mmpose/apis/init.py", line 2, in
from .inference import inference_bottomup, inference_topdown, init_model
File "/home/user/work/test/testdetection/test-env/lib/python3.8/site-packages/mmpose/apis/inference.py", line 10, in
from mmengine.dataset import Compose, pseudo_collate
File "/home/user/work/test/testdetection/test-env/lib/python3.8/site-packages/mmengine/dataset/init.py", line 4, in
from .sampler import DefaultSampler, InfiniteSampler
File "/home/user/work/test/testdetection/test-env/lib/python3.8/site-packages/mmengine/dataset/sampler.py", line 9, in
from mmengine.dist import get_dist_info, sync_random_seed
File "/home/user/work/test/testdetection/test-env/lib/python3.8/site-packages/mmengine/dist/init.py", line 2, in
from .dist import (all_gather_object, all_reduce, all_gather, all_reduce_dict,
File "/home/user/work/test/testdetection/test-env/lib/python3.8/site-packages/mmengine/dist/dist.py", line 15, in
from torch.distributed import ProcessGroup
ImportError: cannot import name 'ProcessGroup' from 'torch.distributed' (/home/user/work/test/testdetection/test-env/lib/python3.8/site-packages/torch/distributed/init.py)

The details of the environment is shown below.
absl-py==1.4.0
addict==2.4.0
aiohttp==3.8.4
aiosignal==1.3.1
async-timeout==4.0.2
attrs==23.1.0
cachetools==5.3.0
certifi==2023.5.7
charset-normalizer==3.1.0
chumpy==0.70
click==8.1.3
cmake==3.26.3
colorama==0.4.6
contourpy==1.0.7
cycler==0.11.0
Cython==0.29.34
distro==1.8.0
filelock==3.12.0
flatbuffers==23.5.9
fonttools==4.39.4
frozenlist==1.3.3
google-auth==2.18.0
google-auth-oauthlib==1.0.0
grpcio==1.54.2
idna==3.4
imageio==2.28.1
importlib-metadata==6.6.0
importlib-resources==5.12.0
jetson-stats==4.2.1
Jinja2==3.1.2
json-tricks==3.16.1
kiwisolver==1.4.4
lazy_loader==0.2
loguru==0.7.0
Markdown==3.4.3
markdown-it-py==2.2.0
MarkupSafe==2.1.2
matplotlib==3.7.1
mdurl==0.1.2
mmcv==2.0.0
mmdet==3.0.0
mmengine==0.7.3
mmpose==1.0.0
mmyolo==0.5.0
model-index==0.1.11
mpmath==1.3.0
multidict==6.0.4
munkres==1.1.4
networkx==3.1
ninja==1.11.1
numpy==1.24.3
oauthlib==3.2.2
onnx==1.8.1
onnx-simplifier==0.3.5
onnxoptimizer==0.3.13
onnxruntime==1.8.0
opencv-python==4.7.0.72
openmim==0.3.7
ordered-set==4.1.0
packaging==23.1
pandas==2.0.1
Pillow==9.5.0
pkg_resources==0.0.0
prettytable==3.7.0
protobuf==4.23.0
pyasn1==0.5.0
pyasn1-modules==0.3.0
pycocotools==2.0.6
Pygments==2.15.1
pyparsing==3.0.9
python-dateutil==2.8.2
pytz==2023.3
PyWavelets==1.4.1
PyYAML==6.0
requests==2.30.0
requests-oauthlib==1.3.1
rich==13.3.5
rsa==4.9
scikit-image==0.20.0
scipy==1.9.1
shapely==2.0.1
six==1.16.0
smbus2==0.4.2
sympy==1.12
tabulate==0.9.0
tensorboard==2.13.0
tensorboard-data-server==0.7.0
termcolor==2.3.0
terminaltables==3.1.10
thop==0.1.1.post2209072238
tifffile==2023.4.12
tomli==2.0.1
torch @ https://developer.download.nvidia.cn/compute/redist/jp/v50/pytorch/torch-1.13.0a0+340c4120.nv22.06-cp38-cp38-linux_aarch64.whl#sha256=2bdbb84f8886cb212ff59b25fa10312006a7a7761bf8926032e6d922f157ced7
torchaudio==2.0.2
torchvision==0.14.0a0+53ba0a4
tqdm==4.65.0
typing_extensions==4.5.0
tzdata==2023.3
urllib3==1.26.15
wcwidth==0.2.6
Werkzeug==2.3.4
xtcocotools==1.13
yacs==0.1.8
yapf==0.33.0
yarl==1.9.2
yolox==0.3.0
zipp==3.15.0

How can I fix this error?
Thanks!