Could you post your setup via: python -m torch.utils.collect_env as well as an executable code snippet to reproduce this issue?
PS: you can post code snippets by wrapping them into three backticks ```, which makes debugging easier.