pytorch 1.4.0 cudnn version code example

Example 1: get pytorch version

python -c "import torch; print(torch.__version__)"

Example 2: check cuda version pytorch

torch.version.cuda