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

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