pytorch之torch.cuda.is_available()——判断GPU可用

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

一般都是在命令行里:
在这里插入图片描述
返回True说明GPU可用