想用GPU跑catboost的机器学习模型,结果报错。找了一下catboost的文档,结果如下:
fit - CatBoostClassifier | CatBoost
Note
Set the task_type
parameter in the class constructor to to train the model on GPU. Training on GPU requires NVIDIA Driver of version 450.xx or higher.
完蛋了,我的电脑显卡是AMD,拜拜了您嘞~~~~
附:NVIDIA和AMD是两家显卡厂商,互为竞争对手关系。