Torch.distributed is not available on Jetson PyTorch

@zergzzlun if you prefer installing the wheel as opposed to using the container, I believe this was the last PyTorch wheel built with USE_DISTRIBUTED enabled:

PyTorch v1.11.0

For newer versions, you would need to build PyTorch from source to enable distributed mode (see this post for build instructions: PyTorch for Jetson)

Also, the reason that you get torch.cuda.is_available = False when you install it with conda, is because those upstream wheels from PyPi/conda/pip were not built for JetPack or with CUDA enabled.