Jetson pytorch cuda cudnn version related logic

Refer this link for more information: