gradientsky/docker-autogluon

AutoML for Image, Text, Time Series, and Tabular Data

Install Instructions | Documentation (Stable | Latest)

AutoGluon automates machine learning tasks enabling you to easily achieve strong predictive performance in your applications. With just a few lines of code, you can train and deploy high-accuracy machine learning and deep learning models on image, text, time series, and tabular data.

Visit auto.gluon.ai for more information.

Image Types

framework - base framework image - contains AutoGluon, runtimes and necessary dependencies
jupyter - image with a jupyter notebook available

Image Tag Naming Scheme

CPU version

1.0.0-cpu-framework-ubuntu20.04-py3.10
 ^     ^   ^          ^          ^
 |     |   type       |          python version
 |     |              |
 |     |              |
 |     |              |
 |     CPU type       linux version
 |
 AutoGluon version

GPU version

1.0.0-cuda11.8-jupyter-ubuntu20.04-py3.10
 ^        ^    ^          ^          ^
 |        |    type       |          python version
 |        |               |
 |        cuda version    |
 |                        |
 |                        linux version
 |
 AutoGluon version

Usage

Start Container and Notebook Server with GPU support

$ docker pull autogluon/autogluon:1.0.0-cuda11.8-jupyter-ubuntu20.04-py3.10
$ docker run --gpus all --shm-size=1G --rm -it -p 8888:8888 \
         autogluon/autogluon:1.0.0-cuda11.8-jupyter-ubuntu20.04-py3.10

Start Container and Notebook Server with CPU-only support

$ docker pull autogluon/autogluon:1.0.0-cpu-jupyter-ubuntu20.04-py3.10
$ docker run --rm --shm-size=1G -it -p 8888:8888 \
         autogluon/autogluon:1.0.0-cpu-jupyter-ubuntu20.04-py3.10

gradientsky / docker-autogluon