@A98 your screenshot shows that you are missing libmpi_cxx.so, have you tried installing libopenmpi-dev from apt like shown in the PyTorch for Jetson thread?
If you continue having problems, you could also try using the l4t-pytorch container, which already includes PyTorch and torchvision pre-installed: