Python package installation

Dependencies:

• numpy
• six
• pandas

To install the Python package:

1. Choose an installation method:

   • pip install
   • conda install
   • Build from source on Linux and macOS
   • Build from source on Windows
   • Build a wheel package

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 available from pip install and conda install have GPU support out-of-the-box.

Devices with compute capability 3.0 and higher are supported in compiled packages.

Training on GPU requires NVIDIA Driver of version 418.xx or higher.

The Python version of CatBoost for CUDA of compute capability 2.0 can be built from source. In this case the following steps are obligatory:

• Step 4 of the Build from source on Linux and macOS operation.
• Step 3 of the Build from source on Windows operation.