install pytorch for cpu code example Example 1: get pytorch version import torch print(torch.__version__) Example 2: check cuda version pytorch torch.version.cuda