安装C-api库
如果从pip
安装的TF不包括C-api库,所以我们需要使用TF的源文件安装TF。
从github上下载git库
git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
因为代码库默认为master
,所以要切换至对应版本r1.13
git checkout r.13
测试r.12
之后的版本
bazel test -c opt -- //tensorflow/... -//tensorflow/compiler/... -//tensorflow/lite/...
配置configure
./configure
bazel build -c opt --copt=-mfpmath=both --copt=-msse4.2 --config=cuda //tensorflow:libtensorflow.so
bazel build -c opt --copt=-mfpmath=both --copt=-msse4.2 --config=cuda //tensorflow:libtensorflow_cc.so
done!