Torch distributed not supported on Orin?

[user75410]

We have an Orin loaded with Jetpack 5.0. We use the following NGC container:
nvcr.io/nvidia/l4t-pytorch: r34.1.1-pth1.12-py3
Looks like Torch is not ‘distributed’

Python 3.8.10 (default, Mar 15 2022, 12:22:08)
[GCC 9.4.0] on linux
Type “help”, “copyright”, “credits” or “license” for more information.

import torch
torch.cuda.is_available()
True
torch.version
‘1.12.0a0+2c916ef.nv22.3’
torch.distributed.is_available()
False

Is there an NGC container available with torch distributed ?

[dusty_nv]

Hi @user75410, you can use nvcr.io/nvidia/l4t-pytorch:r34.1.1-pth1.11-py3 instead, that has distributed enabled.