TensorFlow + MKL 內存泄漏及解決辦法

TensorFlow + MKL 內存泄漏及解決辦法 背景 最近在做深度學習在線推理相關的項目,底層使用TF來作爲推斷框架,我們知道,TF底層是用Eigen完成CPU的計算的,而Eigen的速度只能說一般般,與成熟的計算庫在性能上的差距還是比較明顯的,比如TF支持的intel MKL。 之於如何使用TF+MKL,可以參考此文,其實就是在編譯時加上--config=mkl選項,然後在生成可執行文件時
相關文章
相關標籤/搜索