import torch
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
print(device)
0
点赞
收藏
分享
pytorch 检查GPU可用
相关推荐
0 条评论
import torch
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
print(device)
相关推荐
