A set of base docker images for machine learning.
To pull,
docker pull wqael/mldock:<tag>
Parent images are here: https://github.com/rlan/mldock. Children images are here: https://github.com/rlan/notebooks.
| Tag |
Comment |
Dockerfile |
Info |
latest |
CPU-only. PyTorch 1.4. Keras, TF 1.15.2. |
Dockerfile |
 |
latest-gpu |
CUDA 10.1. PyTorch 1.4. Keras, TF 1.15.2. Requires Nvidia Driver >= 418.39. |
Dockerfile |
 |
sb |
CPU-only. Stable Baselines. PyTorch 1.1. Keras, TF 1.14.0. |
Dockerfile |
 |
sb-gpu |
CUDA 10. Stable Baselines. PyTorch 1.1. Keras, TF 1.14.0. Requires Nvidia Driver >= 410.48. |
Dockerfile |
 |
| Tag (OS-based python) |
Comment |
Dockerfile |
Info |
keras-tf1.12.3-py3-cuda9 |
Nvidia Driver >= 384.xx |
Dockerfile |
 |
keras-tf1.12.3-py2-cuda9 |
Nvidia Driver >= 384.xx |
Dockerfile |
 |
| Tag (Conda-based python) |
Comment |
Dockerfile |
Info |
keras-tf1.14.0-conda3 |
CPU-only |
Dockerfile |
 |
keras-tf1.14.0-conda2 |
CPU-only |
Dockerfile |
 |
keras-tf1.14.0-conda3-cuda10 |
Nvidia Driver >= 410.xx |
Dockerfile |
 |
keras-tf1.14.0-conda2-cuda10 |
Nvidia Driver >= 410.xx |
Dockerfile |
 |
keras-tf1.12.0-conda3-cuda9 |
Nvidia Driver >= 384.xx |
Dockerfile |
 |
keras-tf1.12.0-conda2-cuda9 |
Nvidia Driver >= 384.xx |
Dockerfile |
 |
keras-tf1.4.1-conda3-cuda8 |
Nvidia Driver >= 375.xx |
Dockerfile |
 |
keras-tf1.4.1-conda2-cuda8 |
Nvidia Driver >= 375.xx |
Dockerfile |
 |
| Tag (OS-based python) |
Comment |
Dockerfile |
Info |
pytorch1.8.1 |
CPU-only |
Dockerfile |
 |
pytorch1.8.1-cuda11.1.1 |
Nvidia Driver >= 455.32 |
Dockerfile |
 |
pytorch1.8.1-cuda10.2 |
Nvidia Driver >= 440.33 |
Dockerfile |
 |
pytorch1.7.1-cuda11 |
Nvidia Driver >= 450.36.06 |
Dockerfile |
 |
pytorch1.7.1-cuda101 |
Nvidia Driver >= 418.xx |
Dockerfile |
 |
pytorch1.7.1-cuda92 |
Nvidia Driver >= 396.37 |
Dockerfile |
 |
pytorch1.2-py3-cuda10 |
Nvidia Driver >= 410.xx |
Dockerfile |
 |
pytorch1.2-py2-cuda10 |
Nvidia Driver >= 410.xx |
Dockerfile |
 |
pytorch1.1-py3-cuda9 |
Nvidia Driver >= 384.xx |
Dockerfile |
 |
pytorch1.1-py2-cuda9 |
Nvidia Driver >= 384.xx |
Dockerfile |
 |
pytorch1.0-py3-cuda8 |
Nvidia Driver >= 375.xx |
Dockerfile |
 |
pytorch1.0-py2-cuda8 |
Nvidia Driver >= 375.xx |
Dockerfile |
 |
| Tag (Conda-based python) |
Comment |
Dockerfile |
Info |
pytorch1.3-conda3 |
CPU-only |
Dockerfile |
 |
pytorch1.3-conda3-cuda92 |
Nvidia Driver >= 396.37 |
Dockerfile |
 |
pytorch1.2-conda2 |
CPU-only |
Dockerfile |
 |
pytorch1.2-conda2-cuda10 |
Nvidia Driver >= 410.xx |
Dockerfile |
 |
pytorch1.2-conda3-cuda9 |
Nvidia Driver >= 396.37 |
Dockerfile |
 |
pytorch1.2-conda2-cuda9 |
Nvidia Driver >= 396.37 |
Dockerfile |
 |
pytorch1.1-conda3-cuda9 |
Nvidia Driver >= 384.xx |
Dockerfile |
 |
pytorch1.1-conda2-cuda9 |
Nvidia Driver >= 384.xx |
Dockerfile |
 |
pytorch1.0-conda3-cuda8 |
Nvidia Driver >= 375.xx |
Dockerfile |
 |
pytorch1.0-conda2-cuda8 |
Nvidia Driver >= 375.xx |
Dockerfile |
 |
See deprecated/deprecated.md.
