Learn how to install PyCaret
PyCaret is tested and supported on the following 64-bit systems:
- Python 3.6 – 3.8
- Python 3.9 for Ubuntu only
- Ubuntu 16.04 or later
- Windows 7 or later
Install PyCaret with Python's pip package manager.
pip install pycaret
To install the full version (see dependencies below):
pip install pycaret[full]
In order to avoid potential conflicts with other packages, it is strongly recommended to use a virtual environment, e.g. python3 virtualenv (see python3 virtualenv documentation) or conda environments. Using an isolated environment makes it possible to install a specific version of pycaret and its dependencies independently of any previously installed Python packages.
# create a conda environment
conda create --name yourenvname python=3.8
# activate conda environment
conda activate yourenvname
# install pycaret
pip install pycaret
# create notebook kernel
python -m ipykernel install --user --name yourenvname --display-name "display-name"
With PyCaret, you can train models on GPU and speed up your workflow by 10x. To train models on GPU simply pass
use_gpu = Truein the setup function. There is no change in the use of the API, however, in some cases, additional libraries have to be installed as they are not installed with the default version or the full version. As of the latest release, the following models can be trained on GPU:
- Extreme Gradient Boosting (requires no further installation)
- Catboost (requires no further installation)
- Logistic Regression, Ridge Classifier, Random Forest, K Neighbors Classifier, K Neighbors Regressor, Support Vector Machine, Linear Regression, Ridge Regression, Lasso Regression requires cuML >= 0.15
To install the package directly from GitHub (latest source), use the following command:
pip install git+https://github.com/pycaret/pycaret.git#egg=pycaret
Don't forget to include the
#egg=pycaretpart to explicitly name the project, this way pip can track metadata for it without having to have run the
To launch the test suite, run the following command from outside the source directory:
Docker uses containers to create virtual environments that isolate a PyCaret installation from the rest of the system. PyCaret docker comes pre-installed with a Notebook environment. that can share resources with its host machine (access directories, use the GPU, connect to the Internet, etc.). The PyCaret Docker images are tested for each release.
docker run -p 8888:8888 pycaret/slim
For docker image with full version:
docker run -p 8888:8888 pycaret/full