PyPWDFT:一个轻量级的Python软件,用于单节点10K原子平面波密度功能理论计算
Jun Gao1, Lizhong Fu1, Shizhe Jiao1
1School of Future Technology, Hefei National Laboratory, and Anhui Center for Applied Mathematics, University of Science and Technology of China, Hefei, Anhui 230026, China.
Journal of chemical theory and computation
|February 24, 2025
概括
PyPWDFT是一个用于平面波密度函数理论 (DFT) 计算的Python软件. 它提供与Fortran/C++代码相当的高性能,并使大规模模拟实现高效的CPU-GPU计算.
科学领域:
- 计算物理 计算物理
- 材料科学 材料科学 材料科学
- 量子化学 是一个量子化学.
背景情况:
- 密度函数理论 (DFT) 是现代计算材料科学和量子化学的基石.
- 有效的软件对于处理大规模的DFT计算至关重要,使复杂系统的研究成为可能.
- 现有的高性能DFT代码通常用Fortran或C++编写,这限制了一些研究人员的可访问性.
研究的目的:
- 在Python中开发一个高性能,用户友好的DFT软件包.
- 为了在单个节点上实现大规模的DFT计算,使用CPU-GPU异质计算.
- 为了证明本地Python实现可以实现与传统编译代码可比的性能.
主要方法:
- 使用原生 Python 环境开发 PyPWDFT,利用 NumPy,SciPy 和 CuPy.
- 实现平面波基集和密度函数近似 (局部和混合).
- 与已建立的Fortran/C++ DFT代码进行比较,并评估消费者和专业级GPU的性能.
主要成果:
- PyPWDFT的性能与大型系统 (高达10,000个原子) 的已建立的Fortran/C++ DFT代码相提并论.
- 使用CuPy进行CPU-GPU异质计算,比单线程CPU执行提供了两次级的加快速度.
- 消费级GPU (例如NVIDIA GeForce RTX 4090) 的计算效率与专业级GPU (例如NVIDIA V100) 相美.
结论:
- 作为一个多功能DFT软件包,PyPWDFT显示出巨大的潜力.
- 它的高效性能,可扩展性,数值准确性和跨平台兼容性降低了DFT研究的进入障碍.
- 该软件通过个人电脑上的GUI促进中等规模的DFT计算,扩大了对先进计算方法的访问.
更多相关视频
10:52Multiscale Sampling of a Heterogeneous Water/Metal Catalyst Interface using Density Functional Theory and Force-Field Molecular Dynamics
Published on: April 12, 2019
12.7K
12:11Computation of Atmospheric Concentrations of Molecular Clusters from ab initio Thermochemistry
Published on: April 8, 2020
8.1K
