1.安裝bazel饶唤,根據(jù)官方教程進(jìn)行安裝
- 注意版本號(hào)
- 注意安裝的方式,這里有兩種安裝方式,第一種安裝方法一般是安裝最新版本侈贷,第二種則是手動(dòng)下載所需的版本進(jìn)行安裝bazel version(測(cè)試安裝完成谐檀!)
2.Tensorflow 安裝(一定要從源碼安裝)
官方的第一步也是bazel安裝
這一步一定要進(jìn)行配置抡谐,不然跑不了;這個(gè)可以根據(jù)自己實(shí)際的安裝路徑進(jìn)行配置稚补。
3.實(shí)現(xiàn)簡(jiǎn)單的c++調(diào)用tf的例子
copy里面的測(cè)試代碼
- 訓(xùn)練tf模型童叠,得到一個(gè)pb文件
- 根據(jù)文中的例子,形成以下目錄;這里的cc文件和項(xiàng)目名稱是一致的厦坛,可以不一致不過寫B(tài)UILD文件的時(shí)候要注意了五垮。
- 接下來,在tensorflow repo下(tensorflow的根目錄下面)杜秸,運(yùn)行 ./configure 這一步就是用來設(shè)置一些基本的信息放仗,可以參考別人的配置;也可以根據(jù)自己的實(shí)際情況撬碟。(這步在安裝tensorflow已經(jīng)完成可以不用做)
- bazel build //tensorflow/loader:loader 編譯c++文件
- 從tensorflow的根目錄進(jìn)入到bazel-bin/tensorflow/my_loader文件中
- 把訓(xùn)練生成的pb文件拷貝到該目錄下
- ./my_loader運(yùn)行诞挨,查看結(jié)果