You could try to install the binaries with cudatoolkit=10.2, which should ship with a newer cudnn version, use the NGC container 27, or build PyTorch from source with your local CUDA and cudnn installation.