wqael/mldock
For Dockfiles's, see https://github.com/rlan/mldock
10K+
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.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 |
docker pull wqael/mldock