pytorch install cpu code example

Example 1: get pytorch version

import torch
print(torch.__version__)

Example 2: check cuda version pytorch

torch.version.cuda