WilliamColton
V2EX  ›  问与答

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

  •  
  •   WilliamColton · Aug 1, 2024 · 1310 views
    This topic created in 790 days ago, the information mentioned may be changed or developed.

    关键代码如下:

    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)后识别准确率和速度大幅上升,不加则非常卡而且无法正常识别,今天刚刚上手,不是很清楚怎么回事

    4 replies  •  2024-08-02 09:35:16 +08:00
    ysc3839
        1
    ysc3839  
       Aug 1, 2024 via Android
    改成.eval()呢?
    WilliamColton
        2
    WilliamColton  
    OP
       Aug 1, 2024
    @ysc3839 #1 好像直接开始训练了?但是直接 to(device) 或者没有 to(device) 是不会有这个过程的
    WilliamColton
        3
    WilliamColton  
    OP
       Aug 2, 2024
    @ysc3839 #1 然后每次训练不久就会卡死,调试之后发现好像是被“锁”住了
    ysc3839
        4
    ysc3839  
       Aug 2, 2024 via Android
    About   ·   Help   ·   Advertise   ·   Blog   ·   API   ·   FAQ   ·   Privacy   ·   Solana   ·   2691 Online   Highest 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 38ms · UTC 03:15 · PVG 11:15 · LAX 20:15 · JFK 23:15
    ♥ Do have faith in what you're doing.