PythonOT/POT 库与 NumPy 2.0 兼容性问题解析

问题背景

PythonOT/POT（Python Optimal Transport）是一个用于计算最优传输距离的流行Python库。近期随着NumPy 2.0的发布，许多科学计算库都面临着兼容性挑战，POT库也不例外。

核心问题

当用户在环境中同时安装POT 0.9.3和NumPy 2.0.0时，尝试导入POT库会导致错误。错误信息明确指出：使用NumPy 1.x编译的模块无法在NumPy 2.0环境中运行，可能会崩溃。

技术原因分析

这个问题的根源在于NumPy 2.0引入了二进制兼容性变更。具体来说：

1. NumPy 2.0对C API进行了重大更新，改变了内部数据结构和内存布局
2. POT库的部分核心组件（特别是emd_wrap模块）是用Cython/C++编写的，并针对NumPy 1.x的API进行了编译
3. 这种ABI（应用二进制接口）不兼容导致Python解释器无法正确加载预编译的二进制扩展

解决方案

目前POT开发团队已经在PR #640中解决了这个问题。用户可以选择以下临时解决方案：

1. 降级NumPy：将NumPy版本降级到1.x系列（如1.26.4）

   pip install "numpy<2"
   

2. 等待新版本发布：POT团队即将发布兼容NumPy 2.0的新版本

对用户的影响

这个问题主要影响：

• 使用最新Python 3.12环境的用户
• 需要NumPy 2.0新特性的项目
• 在Windows和Linux平台都会出现

最佳实践建议

对于生产环境用户，建议：

1. 暂时锁定NumPy版本
2. 密切关注POT库的更新
3. 在测试环境中验证新版本兼容性后再进行升级

未来展望

随着科学计算生态系统的演进，这类兼容性问题将逐渐减少。POT团队的快速响应表明开源社区对维护兼容性的重视，这对依赖这些库的研究人员和开发者来说是个积极的信号。