Caffe多线程环境下检测缓慢问题
对于多线程运行环境以及Web框架下(其实也相当于多线程)Caffe运行缓慢的原因可能是仅在某一个线程中设置caffe.set_mode_gpu()。但是该操作不会影响其他线程,此时其他线程还是CPU模式,因此需要对每个线程都要设置GPU模式,即在线程调用的方法中调用一下caffe.set_mode_gpu()即可。
本文来自博客园,作者:NLazyo,转载请注明原文链接:https://www.cnblogs.com/bile/p/9208007.html