Available Images List
InfuseAI provides and maintains certain images which can be base images on Docker hub, here available images are:
TensorFlow
| Framework | Image | CPU / GPU | Python |
|---|---|---|---|
| TensorFlow 2.1 | infuseai/docker-stacks:tensorflow-notebook-tf-v2-3f48358e | CPU | 3.7 |
| TensorFlow 2.1 | infuseai/docker-stacks:tensorflow-notebook-3f48358e-gpu-cuda-10 | GPUCUDA 10.1 | 3.7 |
| TensorFlow 2.1 | infuseai/docker-stacks:tensorflow-notebook-tf-v2-py3.6-8dd7f89a | CPU | 3.6 |
| TensorFlow 2.1 | infuseai/docker-stacks:tensorflow-notebook-py3.6-3f48358e-gpu-cuda-10 | GPUCUDA 10.1 | 3.6 |
| TensorFlow 1.14 | infuseai/docker-stacks:tensorflow-notebook-3f48358e | CPU | 3.7 |
| TensorFlow 1.14 | infuseai/docker-stacks:tensorflow-notebook-3f48358e-gpu | GPUCUDA 9 | 3.7 |
| TensorFlow 1.14 | infuseai/docker-stacks:tensorflow-notebook-py3.6-3f48358e | CPU | 3.6 |
| TensorFlow 1.14 | infuseai/docker-stacks:tensorflow-notebook-py3.6-3f48358e-gpu | GPUCUDA 9 | 3.6 |
PyTorch
| Framework | Image | CPU / GPU | Python |
|---|---|---|---|
| PyTorch 1.4 | infuseai/docker-stacks:pytorch-notebook-3f48358e | CPU | 3.7 |
| PyTorch 1.4 | infuseai/docker-stacks:pytorch-notebook-3f48358e-gpu | GPUCUDA 9 | 3.7 |
| PyTorch 1.4 | infuseai/docker-stacks:pytorch-notebook-py3.6-3f48358e | CPU | 3.6 |
| PyTorch 1.4 | infuseai/docker-stacks:pytorch-notebook-py3.6-3f48358e-gpu | GPUCUDA 9 | 3.6 |
| PyTorch 1.1 | infuseai/docker-stacks:pytorch-notebook-14fb17d1 | CPU | 3.7 |
| PyTorch 1.1 | infuseai/docker-stacks:pytorch-notebook-14fb17d1-gpu | GPUCUDA 9 | 3.7 |
| PyTorch 1.1 | infuseai/docker-stacks:pytorch-notebook-py3.6-8dd7f89a | CPU | 3.6 |
| PyTorch 1.1 | infuseai/docker-stacks:pytorch-notebook-py3.6-8dd7f89a-gpu | GPUCUDA 9 | 3.6 |
Base
| Name | Image | CPU / GPU | Python |
|---|---|---|---|
| AI notebook | infuseai/docker-stacks:ai-notebook-3f48358e | CPU | 3.7 |
| AI notebook | infuseai/docker-stacks:ai-notebook-3f48358e-gpu-cuda-10 | GPUCUDA 10.1 | 3.7 |
| AI notebook | infuseai/docker-stacks:ai-notebook-3f48358e-gpu | GPUCUDA 9 | 3.7 |
| AI notebook | infuseai/docker-stacks:ai-notebook-py3.6-3f48358e | CPU | 3.6 |
| AI notebook | infuseai/docker-stacks:ai-notebook-py3.6-3f48358e-gpu-cuda-10 | GPUCUDA 10.1 | 3.6 |
| AI notebook | infuseai/docker-stacks:ai-notebook-py3.6-3f48358e-gpu | GPUCUDA 9 | 3.6 |