Python package installation


Installation is only supported by the 64-bit version of Python.


  • graphviz (if you want to use plot_tree function)
  • matplotlib
  • numpy (>=1.16.0)
  • pandas (>=0.24)
  • plotly
  • scipy
  • six


Note that in most cases dependencies will be installed automatically using mechanisms built into setuptools, pip or conda.

To install the Python package:

  1. Choose an installation method:

  2. (Optionally) Install additional packages for data visualization support.

  3. (Optionally) Test CatBoost.

Note that there are additional system requirements if training on GPU is required.

GPU system requirements

The versions of CatBoost for Linux and Windows available from pip install and conda install have CUDA-enabled GPU support out-of-the-box.

As of CatBoost 1.2, devices with CUDA compute capability 3.5 - 9.0 are supported in compiled packages.

Training or inference on CUDA-enabled GPU requires NVIDIA Driver of version 450.80.02 or higher.