TensorFlow with MKL for c++环境搭建

本文记录了在C++环境中搭建TensorFlow并利用MKL优化的详细步骤,包括安装特定版本的bazel,从源码编译TensorFlow,解决缺失的libmklml_intel.so和libiomp5.so问题,以及编写和编译测试程序的过程。
摘要由CSDN通过智能技术生成

踩了不少坑,现将正确方法记录如下:

安装bezel


  安装方法参考官方文档
  值得注意的是,目前貌似是由于bazel的bug,tensorflow的1.4版本编译需要用bazel 0.5.4~0.8.0,再往上的版本会编译失败,具体参考tensorflow的这个issues
  另外,bazel 0.12也有问题,具体参看这个issues。因此在这里选择bazel 0.11.1版本。

安装完成后使用命令查看bazel版本

sudo bazel version

显示

Build label: 0.11.1
Build target: bazel-out/k8-opt/bin/src/main/java/com/google/devtools/build/lib/bazel/BazelServer_deploy.jar
Build time: Tue May 14 07:48:23 +50148 (1520362424903)
Build timestamp: 1520362424903
Build timestamp as int: 1520362424903

表明安装成功。

编译TensorFlow源码


从github上下载tensorflow源码,或者切换到要保存代码的目录使用命令

sudo git clone git@github.com:tensorflow/tensorflow.git

然后切换到tensorflow目录,配置编译选项,然后开始build

./configure

这里注意若要编译tensorflow GPU版需要手下安装CUDA以及cuDNN,这里不在赘述。

配置完成之后,先安装tensorflow for python ,如果已经有了python版本可略去这一步

sudo bazel build --config=mkl <
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值