RNN giving CUDNN_STATUS_SUCCESS Error

You can add it into your main script after importing torch.