import torch
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
print(device)
pytorch 检查GPU可用
阅读 11
2023-12-15
相关推荐
精彩评论（0）
0 0 举报
pytorch 检查GPU可用
阅读 11
2023-12-15
import torch
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
print(device)
相关推荐
精彩评论（0）