autogluon/autogluon
AutoGluon: AutoML for Text, Image, and Tabular Data
Image
19
10K+
autogluon/autogluon repository overview
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 dependenciesjupyter- 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
Tag summary
Content type
Image
Digest
sha256:b77aa87f6…
Size
10.6 GB
Last updated
7 months ago
Requires Docker Desktop 4.37.1 or later.