wqael/mldock
For Dockfiles's, see https://github.com/rlan/mldock
Image
1
10K+
wqael/mldock repository overview
MLDock
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.
Tags
Experimental
| 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 |  |
Keras and Tensorflow
| 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 |  |
PyTorch
| Tag (OS-based python) | Comment | Dockerfile | Info |
|---|---|---|---|
pytorch1.5.1 | CPU-only | Dockerfile |  |
pytorch1.5.1-cuda101 | Nvidia Driver >= 418.xx | Dockerfile |  |
pytorch1.5.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 |  |
Older Tags
Tag summary
Content type
Image
Digest
Size
4.3 GB
Last updated
over 5 years ago
docker pull wqael/mldock:pytorch1.8.1-cuda11.1.1