Windows 下编译 AMGX

Windows 下编译 AMGX

在 Windows 下编译 AMGX（NVIDIA 的分布式多 GPU 线性求解器库）需要一些准备工作。以下是详细步骤：

1. 系统要求

• 操作系统: Windows 10 或 11（64 位）
• 编译器: Visual Studio 2019 或 2022（推荐使用 MSVC）
• CUDA Toolkit: 与你的 NVIDIA GPU 驱动兼容的版本（AMGX 通常需要 CUDA 11.0+，最新版本请参考 AMGX GitHub）
• CMake: 3.15 或更高版本
• 依赖项: OpenMP、MPI（可选，用于多节点支持）

2. 安装必要工具

(1) 安装 Visual Studio

• 下载并安装 Visual Studio，勾选 “使用 C++ 的桌面开发” 工作负载。
• 确保安装 MSVC 编译器 和 Windows SDK。

(2) 安装 CUDA Toolkit

• 下载对应版本的 CUDA Toolkit，并安装。
• 安装完成后，验证 nvcc 是否可用：

  nvcc --version
  

(3) 安装 CMake

• 从 CMake 官网 下载并安装最新版本。
• 将 CMake 添加到系统 PATH。

3. 获取 AMGX 源码

git clone https://github.com/NVIDIA/AMGX.git
cd AMGX
git submodule update --init --recursive  # 更新子模块

4. 使用 CMake 生成 Visual Studio 项目

(1) 配置 CMake

• 打开 CMake GUI，设置源码路径（AMGX 根目录）和生成路径（例如 AMGX/build）。
• 点击 Configure，选择对应的 Visual Studio 版本和平台（x64）。
• 根据需要调整 CMake 选项：
  • AMGX_BUILD_SHARED=ON（生成动态库）
  • AMGX_WITH_MPI=ON（如果需要 MPI 支持）
  • CUDA_TOOLKIT_ROOT_DIR（指向 CUDA 安装路径，如 C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.8）

(2) 生成项目

• 点击 Generate，生成 Visual Studio 解决方案（.sln 文件）。

5. 使用 Visual Studio 编译

1. 打开生成的 AMGX.sln（位于 AMGX/build）。
2. 在 Solution Explorer 中：
   • 选择 Release 或 Debug 配置。
   • 右键 Solution ‘AMGX’ → Build Solution。
3. 编译成功后，库文件（.lib 或 .dll）会生成在 build/lib 或 build/bin 目录。

6. 验证安装

• 运行示例程序（如 amgx_capi）：

  cd build/bin
  amgx_capi.exe  # 如果编译了示例
  

• 或编写自己的程序链接 AMGX 库测试。

7. 常见问题

问题 1: CUDA 版本不兼容

• 确保 CUDA 版本与 AMGX 要求的版本匹配（参考 AMGX 文档）。
• 更新 NVIDIA 驱动至最新版本。

问题 2: CMake 找不到 CUDA

• 手动指定 CUDA_TOOLKIT_ROOT_DIR 到 CUDA 安装路径（如 C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.8）。

问题 3: 链接错误

• 确保项目属性中正确配置了 CUDA 库路径（如 cudart.lib）。

8. 参考资源

• AMGX 官方文档: GitHub Wiki
• NVIDIA 开发者论坛: AMGX 讨论区

按照以上步骤操作，你应该能在 Windows 下成功编译 AMGX。如果有其他问题，欢迎进一步提问！