Running AutoSchedule demo meeting error: ImportError: cannot import name 'EarlyStopException' from 'xgboost.core'

When I was running the AutoSchedule demo:

Auto-scheduling a Neural Network for NVIDIA GPU — tvm 0.9.dev0 documentation (apache.org)

I met this bug: ImportError: cannot import name 'EarlyStopException' from 'xgboost.core'

My TVM version is 0.9 dev, and Xgboost version is 1.6.1

What is wrong with the AutoSchedule?

1 Like

It’s version issue of xgboost. see Auto Scheduler (Ansor) XGBoost cost model import error

1 Like

Thank u! It works when I reinstall xgboost 1.5.0

Traceback (most recent call last):
File “/home/lvhang/.local/lib/python3.8/site-packages/xgboost/compat.py”, line 107, in pkg_resources.get_distribution(‘dask’) AttributeError: module ‘pkg_resources’ has no attribute ‘get_distribution’