编译TensorFlow源码启用MPI：configure无相关选项的咨询

首先明确：TensorFlow默认不启用MPI支持，且当前新版本已不再通过./configure交互流程设置MPI选项，需通过Bazel构建参数手动指定相关依赖。

具体操作步骤：

• 先确保系统已安装MPI开发库（如OpenMPI或MPICH），并明确其头文件与库文件的路径
• 执行Bazel编译命令时，添加以下参数：
  • 指定MPI头文件路径：--cxxopt=-I/path/to/mpi/include
  • 指定MPI库文件路径：--linkopt=-L/path/to/mpi/lib
  • 链接MPI库：--linkopt=-lmpi
  • 启用TensorFlow的MPI编译开关：--define=with_mpi=true

举个实际编译示例（假设OpenMPI安装在/usr/lib/openmpi路径下）：

bazel build --config=opt --define=with_mpi=true --cxxopt=-I/usr/lib/openmpi/include --linkopt=-L/usr/lib/openmpi/lib --linkopt=-lmpi //tensorflow/tools/pip_package:build_pip_package

如果使用的是较旧的TensorFlow 1.x系列版本，可在运行./configure前先设置环境变量TF_NEED_MPI=1，这样就能在交互配置选项中看到MPI相关设置项。

内容的提问来源于stack exchange，提问作者Jonathan Ato Markin