Running PyTorch CUDA

I don’t know how that article installs it, but I would recommend trying the l4t-pytorch container for a compatible version of JetPack-L4T that you are running and seeing if the performance is different using the container, because those images come with PyTorch/torchvision already installed and tested using the wheels from this post.