HK560
V2EX  ›  问与答

大概什么水平的显卡能 cuda 加速跑 yolov5 目标识别推理能每秒 30 帧左右?

  •  
  •   HK560 · Apr 25, 2022 · 1861 views
    This topic created in 1519 days ago, the information mentioned may be changed or developed.

    RT

    最近好像很火这个 yolov5 ,自己也试了试。不过因为显卡是 6700xt 没有 cuda ,只能 cpu 算,每秒只有 6 帧左右。

    想着要不要弄个副卡做 cuda 加速,期望能做到 30 帧就可以了,暂时不考虑训练。大概是要什么水平的显卡呢?

    3 replies    2022-04-25 15:41:39 +08:00
    ysc3839
        1
    ysc3839  
       Apr 25, 2022
    PyTorch 有 beta 的 ROCm 支持,不然先试试性能如何?不过只能在 Linux 下用。
    Windows 下还可以试试 ONNX Runtime​​,可以用 DirectML 。
    Liyiw
        2
    Liyiw  
       Apr 25, 2022
    你得说明图片的 size
    HK560
        3
    HK560  
    OP
       Apr 25, 2022 via Android
    @Liyiw 720p
    About   ·   Help   ·   Advertise   ·   Blog   ·   API   ·   FAQ   ·   Solana   ·   1069 Online   Highest 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 33ms · UTC 18:21 · PVG 02:21 · LAX 11:21 · JFK 14:21
    ♥ Do have faith in what you're doing.