检查PyTorch版本及GPU是否安装成功检测

打开终端,切换到pytorch安装环境下,检测程序如下:

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

输出结果:
检查PyTorch版本及GPU是否安装成功检测_第1张图片
表示GPU安装成功,pytorch版本为1.3.0

你可能感兴趣的:(pytorch基础,pytorch,深度学习,机器学习)