What GPU are you running on?
CUDNN_STATUS_ARCH_MISMATCH usually means that your GPU doesn’t support CuDNN.
You can try setting torch.backends.cudnn.enabled to false so that pytorch doesn’t try to use CuDNN and then running your code.