【亲测免费】 NVIDIA Inference Xfer Library (NIXL) 安装与配置指南

1. 项目基础介绍

NVIDIA Inference Xfer Library (NIXL) 是一个旨在加速人工智能推理框架中点对点通信的开源库。它为不同的内存类型（例如 CPU 和 GPU）以及存储（例如文件、块和对象存储）提供抽象层，通过模块化插件架构实现。NIXL 可以与 NVIDIA Dynamo 等推理框架配合使用。

该项目主要使用的编程语言是 C++，同时使用了 Python 和 Shell 脚本进行构建和配置。

2. 项目使用的关键技术和框架

• UCX（Unified Communication Xfer）: 用于高性能通信的库。
• GDRCopy: 用于提高 UCX 性能的工具。
• Meson: 作为构建系统，用于生成构建文件和编译项目。
• pybind11: 用于创建 Python 绑定，以便在 Python 环境中使用 NIXL。

3. 安装和配置准备工作

在开始安装 NIXL 之前，您需要确保系统中已经安装了以下依赖项：

• 构建工具：build-essential（对于 Ubuntu）或 gcc-c++（对于 Fedora）。
• CMake 和 pkg-config。
• Python 3 以及 pip3。
• UCX 库，建议版本为 1.18.0。
• 可选的 GDRCopy，用于提升性能。

Ubuntu 系统的准备工作：

sudo apt update
sudo apt install build-essential cmake pkg-config
sudo apt install python3 python3-pip

Fedora 系统的准备工作：

sudo dnf install gcc-c++ cmake pkg-config
sudo dnf install python3 python3-pip

安装 Python 依赖：

pip3 install meson ninja pybind11

下载并编译 UCX：

wget https://github.com/openucx/ucx/releases/download/v1.18.0/ucx-1.18.0.tar.gz
tar xzf ucx-1.18.0.tar.gz
cd ucx-1.18.0
./configure --enable-shared --disable-static ...
make -j
make -j install-strip
ldconfig

安装 GDRCopy（可选）：

# 请根据实际情况替换 <gdrcopy install> 为正确的安装路径
wget https://github.com/openucx/gdrcopy/releases/download/...
# 解压并编译 GDRCopy
tar xzf gdrcopy-...
cd gdrcopy-...
./configure --with-cuda=<cuda install> ...
make -j
make -j install-strip

安装步骤

1. 克隆 NIXL 仓库到本地：

git clone https://github.com/ai-dynamo/nixl.git
cd nixl

2. 创建构建目录并配置 Meson：

mkdir build && cd build
meson setup ..

3. 编译项目：

ninja

4. 安装项目：

ninja install

5. （可选）构建 Docker 容器：

./contrib/build-container.sh

按照以上步骤操作，您应该能够成功安装和配置 NIXL。如果有任何问题，请参考项目的 README 文件和官方文档获取更多帮助。