构建与安装#
获取源代码#
git clone https://github.com/flagos-ai/FlagCX.git
cd FlagCX
git submodule update --init --recursive
安装#
方式 A — Python 安装(pip install):
pip install . -v --no-build-isolation
方式 B — C++ 库(make):
make <backend>=1 -j$(nproc)
其中 <backend> 为以下选项之一:
USE_NVIDIA: NVIDIA GPU 支持USE_ILUVATAR_COREX: Iluvatar Corex 支持USE_CAMBRICON: 寒武纪支持USE_METAX: MetaX 支持USE_MUSA: 摩尔线程支持USE_KUNLUNXIN: 昆仑芯支持USE_DU: 海光支持USE_ASCEND: 华为昇腾支持USE_AMD: AMD 支持USE_TSM: 清微智能支持USE_ENFLAME: 燧原支持USE_SUNRISE: Sunrise AI 支持USE_GLOO: GLOO 支持USE_MPI: MPI 支持
注意,方式 A 也支持 <backend>=1,允许用户显式指定后端。否则将自动选择。
默认安装路径设置为 build/,您可以手动设置 BUILDDIR 环境变量来自定义构建路径。
您也可以指定 DEVICE_HOME 和/或 CCL_HOME 来指示设备运行时的安装路径和通信库的安装路径。