Caffe多线程环境下检测缓慢问题

 

    对于多线程运行环境以及Web框架下(其实也相当于多线程)Caffe运行缓慢的原因可能是仅在某一个线程中设置caffe.set_mode_gpu()。但是该操作不会影响其他线程,此时其他线程还是CPU模式,因此需要对每个线程都要设置GPU模式,即在线程调用的方法中调用一下caffe.set_mode_gpu()即可。

posted on 2018-06-21 11:21  NLazyo  阅读(348)  评论(0编辑  收藏  举报