Hi,
Sorry that we don’t have a docker with Magama installed. For PyTorch, you can check l4t-pytorch or l4t-ml.
Thanks.