Thanks a lot @ptrblck for looking into it. Below are the details
PyTorch:  1.6.0
torchvision: 0.7.0
cuda : 10.0.130
cudnn: 7.5
GPU: Titan RTX
Installed pytorch using conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
image740×594 17.8 KB