简介:在 Windows 10 系统下,使用 Bazel 从源码编译 TensorFlow 1.15.3 的 pip 安装包,同时支持 CPU 版本并支持 AVX2 指令集。这是一个相对复杂的任务,但以下内容将为您提供一个基本的指导。
在 Windows 10 系统下,使用 Bazel 从源码编译 TensorFlow 1.15.3 的 pip 安装包,同时支持 CPU 版本并支持 AVX2 指令集。这是一个相对复杂的任务,但以下内容将为您提供一个基本的指导。
首先,需要明确以下几点:
这个命令告诉 Bazel 以最优化模式(
bazel build -c opt --copt=-mavx2 --config=win --config=x64 tensorflow/tools/pip package:pip_package
-c opt)进行编译,并使用 AVX2 指令集(--copt=-mavx2)。--config=win 表示 Windows 配置,--config=x64 表示 64 位系统。tensorflow/tools/pip 表示要编译的 TensorFlow 部分,package:pip_package 表示要生成的包类型。bazel-bin/tensorflow/tools/pip_package 目录下生成一个 tensorflow-1.15.3-py3-none-any.whl 文件。这将把 TensorFlow 安装到 Python 中。
pip install --upgrade tensorflow-1.15.3-py3-none-any.whl