CatBoostError: /src/catboost/catboost/cuda/cuda_lib/cuda_buffer.h:542: [0-32] / [0-31]

[animeshkumarpaul]

import catboost as cb

model_params={
    'task_type': 'GPU',
    'devices':'1',
    'loss_function': 'MultiClass',
    'iterations': 1000,
    'early_stopping_rounds': 500,
    'learning_rate' : 0.009,
    'auto_class_weights' : 'SqrtBalanced',
    'l2_leaf_reg' : 3, 'max_leaves':31
    
}

model = cb.CatBoostClassifier(**model_params)

model.fit(train_pool, verbose=True, plot=False, eval_set=valid_pool, use_best_model = True)

I am getting this error:

File ~/.../lib/python3.8/site-packages/catboost/core.py:5131, in CatBoostClassifier.fit(self, X, y, cat_features, text_features, embedding_features, sample_weight, baseline, use_best_model, eval_set, verbose, logging_level, plot, plot_file, column_description, verbose_eval, metric_period, silent, early_stopping_rounds, save_snapshot, snapshot_file, snapshot_interval, init_model, callbacks, log_cout, log_cerr)
5128 if 'loss_function' in params:
5129 CatBoostClassifier._check_is_compatible_loss(params['loss_function'])
-> 5131 self._fit(X, y, cat_features, text_features, embedding_features, None, sample_weight, None, None, None, None, baseline, use_best_model,
5132 eval_set, verbose, logging_level, plot, plot_file, column_description, verbose_eval, metric_period,
5133 silent, early_stopping_rounds, save_snapshot, snapshot_file, snapshot_interval, init_model, callbacks, log_cout, log_cerr)
5134 return self

File ~/.../lib/python3.8/site-packages/catboost/core.py:2357, in CatBoost._fit(self, X, y, cat_features, text_features, embedding_features, pairs, sample_weight, group_id, group_weight, subgroup_id, pairs_weight, baseline, use_best_model, eval_set, verbose, logging_level, plot, plot_file, column_description, verbose_eval, metric_period, silent, early_stopping_rounds, save_snapshot, snapshot_file, snapshot_interval, init_model, callbacks, log_cout, log_cerr)
2353 allow_clear_pool = train_params["allow_clear_pool"]
2355 with log_fixup(log_cout, log_cerr),
2356 plot_wrapper(plot, plot_file, 'Training plots', [_get_train_dir(self.get_params())]):
-> 2357 self._train(
2358 train_pool,
2359 train_params["eval_sets"],
2360 params,
2361 allow_clear_pool,
2362 train_params["init_model"]
2363 )
2365 # Have property feature_importance possibly set
2366 loss = self._object._get_loss_function_name()

File ~/.../lib/python3.8/site-packages/catboost/core.py:1761, in _CatBoostBase._train(self, train_pool, test_pool, params, allow_clear_pool, init_model)
1760 def _train(self, train_pool, test_pool, params, allow_clear_pool, init_model):
-> 1761 self._object._train(train_pool, test_pool, params, allow_clear_pool, init_model._object if init_model else None)
1762 self._set_trained_model_attributes()

File _catboost.pyx:4624, in _catboost._CatBoost._train()
File _catboost.pyx:4673, in _catboost._CatBoost._train()
CatBoostError: /src/catboost/catboost/cuda/cuda_lib/cuda_buffer.h:542: [0-32] / [0-31]

Could you please suggest how to handle this error?