mpi4py-fft -- 使用MPI进行Python的并行快速傅里叶变换 (FFTs)
项目描述
mpi4py-fft
mpi4py-fft 是一个用于计算快速傅里叶变换 (FFTs) 的 Python 包。大数组被分配,并且在幕后由 MPI for Python (mpi4py) 处理通信。为了分配大数组,我们使用了一种 全新的通用算法,该算法允许分配多维数组中的任何索引集。我们可以仅分配一个索引(板块分解)、两个索引集(铅笔分解)或更高维数组中的更多索引。
mpi4py-fft 包含自己的 Python 接口,用于访问串行 FFTW 库。此接口可以使用类似于 pyfftw 的方式使用,甚至可以用于实数到实数的变换,如离散余弦或正弦变换。
更多文档可以在 readthedocs 找到。
安装
可以使用以下命令安装 mpi4py-fft 包
pip install mpi4py-fft
或者,从 GitHub 获取最新版本
pip install git+https://github.com/mpi4py/mpi4py-fft@master
从 coda-forge 通道使用 conda 安装
conda install -c conda-forge mpi4py-fft
或者从主源目录使用 conda build 构建
conda build -c conda-forge conf/ conda create --name mpi4py_fft mpi4py_fft --use-local
这将从 conda-forge 频道中拉取所需的依赖项。
请注意,mpi4py-fft 依赖于 Python 包
mpi4py
numpy
cython
以及串行 C 库
特别是,mpi4py 需要您有一个正常工作的 MPI 安装,包括编译器包装 mpicc,并在搜索路径中。FFTW 的头文件和库也必须在搜索路径中可用,我们将为 FFTW 库中找到的任何精度构建包装器。
上述所有依赖项都可用,并且如果使用 conda 进行安装,将通过 conda-forge 频道下载。然而,pip 无法帮助您解决 MPI 或 FFTW 的问题。
对于 IO,您需要安装支持 MPI 的 h5py 或 netCDF4。两者均可通过 coda-forge 频道获得
conda install -c conda-forge h5py=*=mpi* netcdf4=*=mpi*
项目详情
关闭
mpi4py_fft-2.0.6.tar.gz 的散列
算法 | 散列摘要 | |
---|---|---|
SHA256 | 2d43aba64d1d6098896cb89942c686a7e8832f1ea6f1593e617d70f365407087 |
|
MD5 | f96abc3bed894aa918418c69b34ee56e |
|
BLAKE2b-256 | d87d3cdd7dfae4032dad498311197c4c78040cd16b736739bac6b9c2758fdd7a |