PyTorch-判断pytorch是否支持GPU加速 

import  torch
# 判断pytorch是否支持GPU加速
print (torch.cuda.is_available())
posted @ 2021-07-04 00:00  司砚章  阅读(60)  评论(0)  编辑  收藏  举报