Ubuntu16.04+bazel+tensorflow-gpu1.12

写在前面

本人写的在Win7上配置tensorflow1.12的c++教程WIN7+bazel+VS2015+tensorflow-gpu1.12仅供参考,互相验证

环境

Ubuntu-16.04
bazel-0.15.0

详细步骤

下载tensorflow

git clone https://github.com/tensorflow/tensorflow.git

clone完毕后使用git切换到所需分支即可

git checkout r1.12

或者直接从GitHub官网下载1.12版本的zip包解压。

设置protobuf版本问题

刚开始时没有发现这个问题,踩了很长时间坑,在这mark一下。注:好像1.12版本的不存在这个问题,但没有实际动手实验。
首先打开,记录此处的地址,再找到,修改为之前的地址即可

编译设置

./configure

接下来会进行很多选择,我的配置如下:

开始编译

bazel build --config=opt --config=cuda //tensorflow:libtensorflow_cc.so

如果要编译CPU版的指令如下:

bazel build --config=opt //tensorflow:libtensorflow_cc.so

致谢

参考

https://blog.csdn.net/gubenpeiyuan/article/details/80855644
https://github.com/tensorflow/tensorflow
https://www.tensorflow.org/install/source

未完待续

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。