import torch
# 判断pytorch是否支持GPU加速
print (torch.cuda.is_available())
判断pytorch是否支持GPU加速
最新推荐文章于 2024-08-11 03:12:54 发布
import torch
# 判断pytorch是否支持GPU加速
print (torch.cuda.is_available())