Advertisement

优化编译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等指令集,可按照以下步骤重新进行源代码编译。

  1. 对TensorFlow源代码进行重新编译,直接从GitHub获取源码并执行以下命令:
复制代码
    git clone https://github.com/tensorflow/tensorflow
    
      
    
  1. 在进行源码编译时,需采用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)

还没有任何评论哟~