Caffe在多线程环境或在Python Web框架下运行缓慢可能的原因

    今天搞并发时遇到的问题,纠结许久最终解决,再次记录,希望能帮到需要的人。

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

猜你喜欢

转载自my.oschina.net/kalnkaya/blog/1810471