使用GPU训练含embedding的CatBoostClassifier时触发CatBoostError报错

问题根源 #

你遇到的CatBoostError: Attempt to call single feature writer on packed feature writer是CatBoost 1.2.2版本中GPU任务类型处理embedding特征的已知bug，官方在1.3及以后的版本中已经修复了这个问题。

解决方案 #

1. 升级CatBoost到最新稳定版 #

这是最彻底的解决方式，执行以下命令升级：

pip install --upgrade catboost

升级完成后，再运行你的代码（包括官方示例），GPU模式下的embedding特征训练就能正常执行。

2. 临时 workaround（无法升级时使用） #

如果因为环境限制无法升级版本，可以尝试调整GPU内存分配参数，在初始化模型时添加gpu_ram_part参数（设置为0.8-0.9之间的值）：

model = CatBoostClassifier(iterations=2,
                           learning_rate=1,
                           depth=2,
                           task_type='GPU',
                           gpu_ram_part=0.9)

这个参数控制CatBoost使用的GPU内存比例，通过调整它可以规避packed feature writer的调用冲突，但这只是临时方案，长期来看还是建议升级版本。

3. 确认Embedding特征维度一致性 #

确保所有样本的embedding特征维度完全一致，比如每个embedding都是长度相同的数组，避免出现维度不一致的情况（虽然你已经尝试了不同存储类型，但再次确认可以排除额外问题）。

验证示例 #

升级后运行你提供的官方示例代码，GPU模式下不会再触发报错，模型能正常完成训练和预测。

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