Linux 系统编译安装 GPU 版 LAMMPS 指南

问题二：找不到 MPI 库

错误信息：

Could NOT find MPI (missing: MPI_CXX_FOUND CXX)

原因：未安装 OpenMPI 或 CMake 无法找到 MPI。

解决方案：

# 安装 OpenMPI
sudo apt install -y libopenmpi-dev openmpi-bin
# 或禁用 MPI（仅限单机测试）
cmake ... -DBUILD_MPI=no ...

问题三：无法检测 GPU 架构

错误信息：

CUDA enabled but no NVIDIA GPU architecture currently enabled and auto-detection failed. Please give one -DKokkos_ARCH_{..}=ON' to enable an NVIDIA GPU architecture.

原因：CMake 无法自动检测 GPU 计算能力，需要手动指定。

解决方案：根据 GPU 型号手动指定架构

常见 GPU 架构对照表：

查询 GPU 计算能力：

# 查看 GPU 型号和计算能力
nvidia-smi --query-gpu=name,compute_cap --format=csv
# 或编译测试程序
cd /tmp
cat > test_gpu.cu <<'EOF'
#include <stdio.h>
int main() {
  cudaDeviceProp prop;
  cudaGetDeviceProperties(&prop, 0);
  printf("GPU: %s\n", prop.name);
  printf("Compute Capability: %d.%d\n", prop.major, prop.minor);
  return 0;
}
EOF
nvcc test_gpu.cu -o test_gpu && ./test_gpu

完整安装流程

基于上述问题，以下是完整的安装步骤：

步骤 1：环境准备

# 安装所有依赖
sudo apt update
sudo apt install -y build-essential cmake git wget gfortran
sudo apt install -y gcc-12 g++-12 # 与 CUDA 12.2 兼容
sudo apt install -y libopenmpi-dev openmpi-bin
# 验证安装
gcc-12 --version
g++-12 --version
mpirun --version
nvidia-smi

步骤 2：下载并解压 LAMMPS

wget https://download.lammps.org/tars/lammps-stable.tar.gz
tar -xzvf lammps-stable.tar.gz
cd lammps-*

步骤 3：创建构建目录

mkdir build
cd build

步骤 4：CMake 配置

针对 NVIDIA A100 的配置：

cmake -C../cmake/presets/basic.cmake \
  -C../cmake/presets/kokkos-cuda.cmake \
  -DCMAKE_C_COMPILER=gcc-12 \
  -DCMAKE_CXX_COMPILER=g++-12 \
  -DPKG_KOKKOS=yes \
  -DKokkos_ENABLE_CUDA=yes \
  -DKokkos_ARCH_AMPERE80=ON \
  -DKokkos_ENABLE_OPENMP=yes \
  -DBUILD_MPI=yes \
  -DFFT_KOKKOS=CUFFT \
  ../cmake

关键参数说明：

步骤 5：编译安装

# 并行编译（使用所有 CPU 核心）
make -j$(nproc)
# 安装（可选）
sudo make install

编译过程可能需要 10-30 分钟，取决于系统性能。

验证与运行

1. 验证安装

# 检查 LAMMPS 版本和编译选项
./lmp -h | head -20
# 应看到 KOKKOS 和 CUDA 相关的包已启用

2. 运行测试

单 GPU 运行：

./lmp -k on g 1 -sf kk -in ../bench/in.lj

参数说明：

• -k on：启用 KOKKOS
• g 1：使用 1 个 GPU
• -sf kk：自动为支持的命令添加 kk 后缀，启用 KOKKOS 加速

多 GPU + MPI 运行：

mpirun -np4 ./lmp -k on g 4 -sf kk -in input.script

GPU + OpenMP 混合并行：

mpirun -np1 ./lmp -k on g 1 t 8 -sf kk -in input.script

• t 8：使用 8 个 OpenMP 线程

3. 性能对比

可以通过以下方式对比 CPU 和 GPU 性能：

# CPU 版本（仅 MPI）
mpirun -np8 ./lmp -in input.script
# GPU 版本（KOKKOS）
mpirun -np1 ./lmp -k on g 1 -sf kk -in input.script

总结

本文详细介绍了在 Linux 系统上安装 GPU 版本 LAMMPS 的完整流程，总结关键要点如下：

关键步骤回顾

1. 环境准备：安装 GCC 12（与 CUDA 兼容）、OpenMPI、CMake 等依赖
2. 架构指定：根据 GPU 型号手动指定 Kokkos_ARCH_* 参数
3. 编译配置：使用 KOKKOS 包启用 CUDA 加速
4. 运行优化：合理使用 -k on g {n} 和 -sf kk 参数

常见问题速查

后续优化建议

1. 性能调优：根据体系大小调整 GPU 线程数和 MPI 进程数
2. 多节点运行：在集群环境中结合 MPI 实现多节点 GPU 并行
3. 监控工具：使用 nvidia-smi 监控 GPU 利用率和内存使用

通过 GPU 加速，LAMMPS 的分子动力学模拟可以获得数倍甚至数十倍的性能提升，显著缩短科研计算时间。

参考资源

• LAMMPS 官方文档
• KOKKOS 编程指南
• NVIDIA CUDA 文档