Are you using the latest PyTorch release (1.7.1) and if not could update to it and rerun your script?
If you are already on the latest version, could you post a minimal code snippet to reproduce this issue and post your current setup, i.e. used GPU, CUDA, cudnn version etc.?