Installation

Note

AutoGluon requires Python version 3.6 or 3.7. Linux is the only operating system fully supported for now (complete Mac OSX and Windows versions will be available soon).

Select your preferences below and run the corresponding install commands:

OS: Linux Mac

Linux.
Mac OSX.

Version: PIP Source

PIP Release.
Install AutoGluon from source.

Backend: CPU GPU

Build-in backend for CPU.
Required to run on Nvidia GPUs.

Command:

pip install --upgrade mxnet
pip install autogluon
# Here we assume CUDA 10.0 is installed.  You should change the number
# according to your own CUDA version (e.g. mxnet-cu101 for CUDA 10.1).
pip install --upgrade mxnet-cu100
pip install autogluon
pip install --upgrade mxnet
git clone https://github.com/awslabs/autogluon
cd autogluon && python setup.py develop
# Here we assume CUDA 10.0 is installed.  You should change the number
# according to your own CUDA version (e.g. mxnet-cu101 for CUDA 10.1).
pip install --pre --upgrade mxnet-cu100
git clone https://github.com/awslabs/autogluon
cd autogluon && python setup.py develop

Note

If you don’t have them, please first install: XCode, Homebrew, LibOMP. Once you have Homebrew, LibOMP can be installed via:

brew install libomp
pip install --upgrade mxnet
pip install autogluon

Note

AutoGluon is not yet fully functional on Mac OSX. If you encounter MXNet system errors, please use Linux instead. However, you can currently use AutoGluon for less compute-intensive TabularPrediction tasks on your Mac laptop (but only with hyperparameter_tune = False).

Note

GPU usage is not yet supported on Mac OSX, please use Linux to utilize GPUs in AutoGluon.

Note

If you don’t have them, please first install: XCode, Homebrew, LibOMP. Once you have Homebrew, LibOMP can be installed via:

brew install libomp
pip install --upgrade mxnet
git clone https://github.com/awslabs/autogluon
cd autogluon && python setup.py develop

Note

AutoGluon is not yet fully functional on Mac OSX. If you encounter MXNet system errors, please use Linux instead. However, you can currently use AutoGluon for less compute-intensive TabularPrediction tasks on your Mac laptop (but only with hyperparameter_tune = False).

Note

GPU usage is not yet supported on Mac OSX , please use Linux to utilize GPUs in AutoGluon.

Next steps