Available Images List

Enterprise Applicable to Enterprise Edition

Community Applicable to Community Edition

InfuseAI provides and maintains certain images which can be used as base images on infuseai/docker-stacks↗ on Docker Hub, here available images are:

TensorFlow

JupyterLab v2 with PrimeHub Extension

Framework	Image	CPU / GPU	Python
TensorFlow 2.5	infuseai/docker-stacks:tensorflow-notebook-v2-5-0-1eb85fe1	CPU	3.7
TensorFlow 2.5	infuseai/docker-stacks:tensorflow-notebook-v2-5-0-1eb85fe1-gpu-cuda-11	GPU CUDA 11	3.7
TensorFlow 2.4	infuseai/docker-stacks:tensorflow-notebook-v2-4-1-e7725c59	CPU	3.7
TensorFlow 2.4	infuseai/docker-stacks:tensorflow-notebook-v2-4-1-e7725c59-gpu-cuda-11	GPU CUDA 11	3.7
TensorFlow 1.15	infuseai/docker-stacks:tensorflow-notebook-v1-15-4-1eb85fe1	CPU	3.7
TensorFlow 1.15	infuseai/docker-stacks:tensorflow-notebook-v1-15-4-1eb85fe1-gpu	GPU	3.7
TensorFlow 1.14	infuseai/docker-stacks:tensorflow-notebook-b3c52f35	CPU	3.7
TensorFlow 1.14	infuseai/docker-stacks:tensorflow-notebook-b3c52f35-gpu	GPU	3.7

---

PyTorch

JupyterLab v2 with PrimeHub Extension

Framework	Image	CPU / GPU	Python
PyTorch 1.8	infuseai/docker-stacks:pytorch-notebook-v1-8-0-1eb85fe1	CPU	3.7
PyTorch 1.8	infuseai/docker-stacks:pytorch-notebook-v1-8-0-1eb85fe1-gpu-cuda-11	GPU CUDA 11	3.7
PyTorch 1.7	infuseai/docker-stacks:pytorch-notebook-v1-7-0-a8109814	CPU	3.7
PyTorch 1.7	infuseai/docker-stacks:pytorch-notebook-v1-7-0-a8109814-gpu-cuda-11	GPU CUDA 11	3.7
PyTorch 1.7	infuseai/docker-stacks:pytorch-notebook-v1-7-0-cuda10-2f118efb-gpu-cuda-10	GPU CUDA 10	3.7
PyTorch 1.5	infuseai/docker-stacks:pytorch-notebook-b3c52f35	CPU	3.7
PyTorch 1.5	infuseai/docker-stacks:pytorch-notebook-b3c52f35-gpu	GPU CUDA 9	3.7

---

Base

JupyterLab v2 with PrimeHub Extension

Name	Image	CPU / GPU	Python
AI notebook	infuseai/docker-stacks:ai-notebook-1eb85fe1	CPU	3.7
AI notebook	infuseai/docker-stacks:ai-notebook-1eb85fe1-gpu	GPU	3.7
AI notebook	infuseai/docker-stacks:ai-notebook-1eb85fe1-gpu-cuda-11	GPU CUDA 11	3.7