用C++導(dǎo)入tensorflow模型環(huán)境配置

1.安裝bazel,根據(jù)官方教程進(jìn)行安裝

  • 注意版本號
  • 注意安裝的方式,這里有兩種安裝方式,第一種安裝方法一般是安裝最新版本,第二種則是手動下載所需的版本進(jìn)行安裝bazel version(測試安裝完成!)

2.Tensorflow 安裝(一定要從源碼安裝)

官方的第一步也是bazel安裝


官網(wǎng)截圖
配置configure
配置configure

這一步一定要進(jìn)行配置,不然跑不了;這個可以根據(jù)自己實際的安裝路徑進(jìn)行配置。

安裝tensorflow

3.實現(xiàn)簡單的c++調(diào)用tf的例子

copy里面的測試代碼

  • 訓(xùn)練tf模型,得到一個pb文件
  • 根據(jù)文中的例子,形成以下目錄;這里的cc文件和項目名稱是一致的,可以不一致不過寫B(tài)UILD文件的時候要注意了。
目錄
  • 接下來,在tensorflow repo下(tensorflow的根目錄下面),運行 ./configure 這一步就是用來設(shè)置一些基本的信息,可以參考別人的配置;也可以根據(jù)自己的實際情況。(這步在安裝tensorflow已經(jīng)完成可以不用做)
  • bazel build //tensorflow/loader:loader 編譯c++文件
  • 從tensorflow的根目錄進(jìn)入到bazel-bin/tensorflow/my_loader文件中
  • 把訓(xùn)練生成的pb文件拷貝到該目錄下
  • ./my_loader運行,查看結(jié)果
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容