Hi @john.ojanen, was this using the l4t-pytorch container, or installing the PyTorch wheel outside of container? Feel free to open a new topic about installing PyTorch/torchvision if you’re having problems with it - I recommend following the install instructions from this topic: PyTorch for Jetson. The version of CUDA is the same between JetPack 5.1 and JetPack 5.1.1 (CUDA 11.4.315)