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
|