It also looks like you are using a different build of PyTorch built for Python 3.9 that was installed through Conda.
Instead, please install one of the PyTorch pip wheels from this post (these were built with CUDA/cuDNN enabled) or use the l4t-pytorch container.