Which nvcr.io/nvidia/l4t-pytorch images have torch.distributed enabled?

I’m running Jetpack 5.0.2 (l4t R35.1.0), and need a docker image with torch distributed enabled.
I’ve found the repo, GitHub - dusty-nv/jetson-containers: Machine Learning Containers for NVIDIA Jetson and JetPack-L4T, but is there a way to tell if a given image is compiled with torch.distributed enabled?

Hi @niklaspandersson, I believe the nvcr.io/nvidia/l4t-pytorch:r35.1.0-pth1.11-py3 container image should have torch.distributed enabled.

1 Like