python检查PyTorch版本及cuda是否安装成功

import torch
print(torch.__version__)
print(torch.cuda.is_available())

 输出True 就代表安装成功,反之安装失败

1.7.0
True

你可能感兴趣的:(安装,python,pytorch,人工智能)