Good to hear, but that’s not really a solution.
Could you post which PyTorch, CUDA and cuDNN versions you are using:
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())

so that I could try to reproduce this issue?