进入python当中:
执行
>>>import torch
>>>import trochvision
>>>print(torch.cuda.is_available())
True
本文标题:验证torch的GPU是否成功
本文链接:https://www.haomeiwen.com/subject/bgfnnftx.html
网友评论