facing the same issue

GPU:1080Ti
Cuda: 10.1 (nvcc -V)

tried pytorch 1.6, 1.7.0+cu101, 1.7.1+cu101
The same settings were working previously but after NVIDIA driver update to 460.39 it stopped working.