Back to Catboost

{{ python-package }} installation

catboost/docs/en/concepts/python-installation.md

1.2.102.3 KB
Original Source

{{ python-package }} installation

{% include supported-versions %}

{% note alert %}

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

{% endnote %}

Dependencies:

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

{% note info %}

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

{% endnote %}

To install the {{ python-package }}:

  1. Choose an installation method:

  2. {% include general-install-data-visualization-support-package %}

  3. (Optionally) Additional setup if user-defined functions are used:

    {% include python__user-defined-functions-dependencies %}

  4. (Optionally) Test {{ product }}.

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

GPU system requirements {#gpu-system-requirements}

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

{% include installation-compute-capability-requirements %}

{% include installation-nvidia-driver-reqs %}