Caffe多線程環境下檢測緩慢問題

 

    對於多線程運行環境以及Web框架下(其實也至關於多線程)Caffe運行緩慢的緣由多是僅在某一個線程中設置caffe.set_mode_gpu()。可是該操做不會影響其餘線程,此時其餘線程仍是CPU模式,所以須要對每一個線程都要設置GPU模式,即在線程調用的方法中調用一下caffe.set_mode_gpu()便可。多線程

相關文章
相關標籤/搜索