import  torch
# 判断pytorch是否支持GPU加速
print (torch.cuda.is_available())
转载请注明出处,欢迎讨论和交流!