关于 YOLO v10 使用过程中的小问题求教一下各位 v 友

查看 21|回复 0
作者:WilliamColton   
关键代码如下:
device = 'cuda' if torch.cuda.is_available() else 'cpu'
print(f"Using device: {device}")
# 加载预训练的 YOLOv10n 模型并将其移到 GPU (如果可用)
model = YOLO("yolov10n.pt").to(device)
部分输出如下:
Using device: cpu
明明都没有使用 GPU ,但是末尾处加 to(device)后识别准确率和速度大幅上升,不加则非常卡而且无法正常识别,今天刚刚上手,不是很清楚怎么回事
您需要登录后才可以回帖 登录 | 立即注册

返回顶部