catboost: GPU via python error

Your example:

import numpy
from catboost import CatBoostRegressor

dataset = numpy.array([[1,4,5,6],[4,5,6,7],[30,40,50,60],[20,15,85,60]])
train_labels = [1.2,3.4,9.5,24.5]
model = CatBoostRegressor(verbose=True, device_type='GPU')
fit_model = model.fit(dataset, train_labels)

print(fit_model.get_params())

---

CatboostError Traceback (most recent call last) <ipython-input-25-3e54dc567647> in <module>() 5 train_labels = [1.2,3.4,9.5,24.5] 6 model = CatBoostRegressor(verbose=True, device_type=‘GPU’) ----> 7 fit_model = model.fit(dataset, train_labels) 8 9 print(fit_model.get_params())

/home/alex/catboost/catboost/python-package/catboost/core.py in fit(self, X, y, cat_features, sample_weight, baseline, use_best_model, eval_set, verbose, plot) 500 model : CatBoost 501 “”" –> 502 return self._fit(X, y, cat_features, sample_weight, baseline, use_best_model, eval_set, verbose, plot) 503 504 def _predict(self, data, prediction_type, ntree_start, ntree_end, thread_count, verbose):

/home/alex/catboost/catboost/python-package/catboost/core.py in _fit(self, X, y, cat_features, sample_weight, baseline, use_best_model, eval_set, verbose, plot) 454 raise ImportError(str(e)) 455 with log_fixup(): –> 456 self._train(X, eval_set, params) 457 if calc_feature_importance: 458 setattr(self, “_feature_importance”, self.get_feature_importance(X))

_catboost.pyx in _catboost._CatBoostBase._train (/home/alex/.ya/build/build_root/nzps/0003ca/catboost/python-package/catboost/_catboost.pyx.cpp:15678)()

_catboost.pyx in _catboost._CatBoost._train (/home/alex/.ya/build/build_root/nzps/0003ca/catboost/python-package/catboost/_catboost.pyx.cpp:10531)()

_catboost.pyx in _catboost._CatBoost._train (/home/alex/.ya/build/build_root/nzps/0003ca/catboost/python-package/catboost/_catboost.pyx.cpp:10329)()

CatboostError: catboost/cuda/train_lib/json_options.h:415: Unknown param: rsm

And i have no idea what to do next.

I’ve just installed cuda-9-0-local.