导入dask_cuda时出现parse_memory_limit导入错误求助
解决dask_cuda导入时的ImportError: cannot import name 'parse_memory_limit' from 'distributed.worker'
问题原因
你的dask-cuda版本(22.6.0)与当前安装的dask/distributed版本(2021.12.0)不兼容。dask-cuda 22.6.0依赖的distributed版本已将parse_memory_limit函数迁移到distributed.worker模块,但你使用的2021.12.0版本中该函数仍位于distributed.utils模块,因此触发导入错误。
解决方案
有两种可行的修复方式:
方案1:升级dask和distributed到兼容版本
将dask和distributed升级到与dask-cuda 22.6.0匹配的22.6.x系列版本:
# 使用conda升级(推荐) conda install dask distributed=22.6 -c conda-forge # 或者使用pip升级 pip install --upgrade dask distributed==22.6.0
方案2:降级dask-cuda到与当前dask/distributed兼容的版本
将dask-cuda降级到21.12.0版本(与你的dask/distributed 2021.12.0版本对齐):
pip install dask-cuda==21.12.0
验证
执行以下代码确认修复成功:
from dask_cuda import LocalCUDACluster from dask.distributed import Client
内容的提问来源于stack exchange,提问作者shambakey1
相关产品推荐
相关产品推荐

