TSINGSEE青犀视频AI智能识别功能开发如何通过GPU实现加速识别?

大家知道我们最近在EasyCVR的AI智能技术方面下了比较大的功夫,近期一直在对智能识别做适配上的测试,待测试完毕即可上线,大家可以关注我们,获取我们最新的研发进度和测试进度。

由于AI识别是一个消耗工程量较大的进程,如果使用CPU进行AI识别则不会加速,并且还会产生非常卡顿的现象,效果很不理想,所以我们在考虑能否通过gpu加速AI识别。

首先查看电脑有没有gpu。打开任务管理器,并点击性能,最下面有GPU的信息,如果电脑有GPU的话,说明电脑支持GPU。

下一步需要查看电脑中GPU的信息。具体操作是先在桌面右击找到NVIDIA控制面板,并点击出现下图画面:

进入NVIDIA控制面板之后,点击左下角的系统信息。进入这个系统信息之后,我们就会看到相关的显卡信息。右边就是显卡的详细信息。

找到显卡的详细信息之后,我们就可以安装对应的python torch使用GPU来加速识别。不过在此之前要先安装英伟达的CUDA Toolkit来使用GPU加速,否则无法使用GPU。

以下是CUDA Toolkit的版本兼容:

原文地址:https://www.cnblogs.com/TSINGSEE/p/15157669.html