大概什么水平的显卡能 cuda 加速跑 yolov5 目标识别推理能每秒 30 帧左右？
HK560 · 2022-04-25 10:44:38 +08:00 · 1268 次点击这是一个创建于 942 天前的主题，其中的信息可能已经有所发展或是发生改变。
RT
最近好像很火这个 yolov5 ，自己也试了试。不过因为显卡是 6700xt 没有 cuda ，只能 cpu 算，每秒只有 6 帧左右。
想着要不要弄个副卡做 cuda 加速，期望能做到 30 帧就可以了，暂时不考虑训练。大概是要什么水平的显卡呢？
 |
1
ysc3839 2022-04-25 11:06:02 +08:00
PyTorch 有 beta 的 ROCm 支持，不然先试试性能如何？不过只能在 Linux 下用。
Windows 下还可以试试 ONNX Runtime，可以用 DirectML 。 |
 |
2
Liyiw 2022-04-25 11:49:39 +08:00
你得说明图片的 size
|
Select Language