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

 

    对于多线程运行环境以及Web框架下(其实也至关于多线程)Caffe运行缓慢的缘由多是仅在某一个线程中设置caffe.set_mode_gpu()。可是该操做不会影响其余线程,此时其余线程仍是CPU模式,所以须要对每一个线程都要设置GPU模式,即在线程调用的方法中调用一下caffe.set_mode_gpu()便可。多线程

相关文章
相关标签/搜索