Could you run your code on the CPU and check, if it’s working or yields any error message?
In case it’s working, could you post an executable code snippet as well as your PyTorch, CUDA, and cudnn version?