Thank you for your response. I have another question, I have installed pytorch with pip install torch==1.2.0 torchvision==0.4.0 does it come with cuDNN installation?
torch.backends.cudnn.version() gives me 7602 which make me believe that it has cudnn version 7.6 with cuda version ‘10.0.130’. If that’s the case, how can I  change the cuDNN version?
Thank you for the help.