优化编译TensorFlow 1.4代码支持SSE-AVX-FMA指令集 (Python 3.5版本)
发布时间
阅读量:
阅读量
TensorFlow在运行过程中显示如下提示信息,表明当前工作站具备相应的指令集支持,但在编译阶段并未将其纳入。
W tensorflow/core/platform/cpu_feature_guard.cc:45] TensorFlow库在编译时未启用SSE4.1指令集,而该指令集在当前设备中是可用的,并且能够提升CPU计算效率。
若需使TensorFlow支持SSE/AVX/FMA等指令集,可按照以下步骤重新进行源代码编译。
- 对TensorFlow源代码进行重新编译,直接从GitHub获取源码并执行以下命令:
git clone https://github.com/tensorflow/tensorflow
- 在进行源码编译时,需采用bazel工具,首要步骤是完成相关依赖库的安装
sudo apt-get install python3-numpy python3-dev python3-pip python3-wheel libcupti-dev
引入bazel源
echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8"
全部评论 (0)
还没有任何评论哟~
