SageMath环境下运行Dask涉及非纯Python代码时报ImportError

问题根因 #

这个报错的核心原因是Dask默认拉起的worker进程使用了系统普通Python解释器，而非Sage封装的专属Python运行环境。Sage的大量Cython扩展模块依赖Sage自身初始化后的路径配置、全局运行上下文，普通Python环境没有完成Sage的环境初始化，自然无法导入complex_mpfr这类Sage专属的扩展模块。

你写的1+1触发报错的原因也很简单：Sage环境下的整数字面量默认会被转换为sage.rings.integer.Integer类型，这段运算逻辑涉及Sage专属对象的创建与运算，序列化后传到worker端反序列化时需要加载Sage的整数相关模块，自然会触发导入失败；去掉该逻辑后代码是纯Python实现，不需要加载Sage模块，所以能正常运行。

解决方案 #

方案1：本地集群场景直接指定worker使用的Python解释器 #

首先在Sage交互环境中执行以下命令获取Sage对应Python解释器的绝对路径：

which python

你当前环境下返回的路径大概率是~/.sage/local/bin/python，替换到Client初始化的参数中即可：

from dask import delayed
from dask.distributed import Client

# 替换python_executable参数值为上一步拿到的实际路径
client = Client(n_workers=4, python_executable="~/.sage/local/bin/python")

def Hello():
    1+1
    return 'Hello World'

z = delayed(Hello)()
z.compute()

方案2：手动部署集群场景统一使用Sage Python启动组件 #

如果是需要手动启动调度器和worker的分布式场景，全程使用Sage的Python解释器启动Dask组件即可：

• 启动调度器：sage -python -m distributed.cli.dask_scheduler
• 启动worker：sage -python -m distributed.cli.dask_worker 调度器地址:端口

补充优化 #

如果调整解释器后还有偶现的导入异常，可以在所有提交给Dask执行的函数开头导入Sage全量环境，确保Sage的运行上下文被完整初始化：

def Hello():
    import sage.all
    1+1
    return 'Hello World'

内容的提问来源于stack exchange，提问作者Sam Ballas