I have the same issue. Here are the versions I have:
pytorch       1.6.0        py3.7_cuda10.1.243_cudnn7.6.3_0    pytorch
cudatoolkit   10.1.243     h6bb024c_0

NVCC:
Cuda compilation tools, release 10.1, V10.1.243
NVIDIA-SMI:
| NVIDIA-SMI 418.126.02   Driver Version: 418.126.02   CUDA Version: 10.1|