import torch
# 当前设备
current_device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print("当前设备: ", current_device)
发现cuda来生成json文件很快,只花了6分钟,而CPU还没生成json文件
import torch
# 当前设备
current_device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print("当前设备: ", current_device)
发现cuda来生成json文件很快,只花了6分钟,而CPU还没生成json文件