Opencv4.2 DNN模塊 終於支持Nividia GPU了!

      最近在做一個項目,用opencv 加載訓練好的caffe模型進行評估,速度很慢,於是打開nividia-smi 查看GPU狀態,發現GPU沒有跑起來。通過一翻查閱,原來只有最新版的Opencv4.2才支持英偉達顯卡進行加速,而且需要對源碼進行配置,編譯,才能使用。

在RTX 2080 Ti 上跑的性能結果如下:

具體的編譯過程可以參考如下鏈接:

1.https://cuda-chen.github.io/image%20processing/programming/2020/02/22/build-opencv-dnn-module-with-nvidia-gpu-support-on-ubuntu-1804.html

2.https://www.pyimagesearch.com/2020/02/03/how-to-use-opencvs-dnn-module-with-nvidia-gpus-cuda-and-cudnn/

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章