Hello, there is a NGC Pytorch container which has support for both arm64 and amd64. This should support the devkit with dGPU. For determining which version of the container to pull, we can reference Frameworks Support Matrix - NVIDIA Docs and the dGPU base image for the Holoscan Docker image (in v0.6) nvcr.io/nvidia/tensorrt:22.03-py3, so you could try nvcr.io/nvidia/pytorch:22.03-py3.