检查pytorch是否有gpu

>>> import torch
>>> torch.cuda.is_available()

True

posted @ 2023-03-09 16:50  bitterteaer  阅读(12)  评论(0)  编辑  收藏  举报