InfuseAI Images List

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

Framework	Image	CPU / GPU	Python
TensorFlow 2.13	infuseai/docker-stacks:tensorflow-notebook-v2-13-0-1951c2b3	CPU	3.9
TensorFlow 2.13	infuseai/docker-stacks:tensorflow-notebook-v2-13-0-1951c2b3-gpu-cuda-12	GPU CUDA 12	3.9
TensorFlow 2.8	infuseai/docker-stacks:tensorflow-notebook-v2-8-0-6dec5b2e	CPU	3.7
TensorFlow 2.8	infuseai/docker-stacks:tensorflow-notebook-v2-8-0-6dec5b2e-gpu-cuda-11	GPU CUDA 11	3.7
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

Framework	Image	CPU / GPU	Python
PyTorch v2.0.0	infuseai/docker-stacks:pytorch-notebook-v2-0-0-1951c2b3	CPU	3.9
PyTorch v2.0.0	infuseai/docker-stacks:pytorch-notebook-v2-0-0-1951c2b3-gpu-cuda-12	GPU CUDA 12	3.9
PyTorch 1.10.2	infuseai/docker-stacks:pytorch-notebook-v1-10-2-6dec5b2e	CPU	3.7
PyTorch 1.10.2	infuseai/docker-stacks:pytorch-notebook-v1-10-2-6dec5b2e-gpu-cuda-11	GPU CUDA 11	3.7
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

Name	Image	CPU / GPU	Python
SciPy	infuseai/docker-stacks:scipy-notebook-1951c2b3	CPU	3.9
SciPy	infuseai/docker-stacks:scipy-notebook-1951c2b3-gpu-cuda-12	GPU CUDA 12	3.9
Base notebook	infuseai/docker-stacks:base-notebook-1951c2b3	CPU	3.9
Base notebook	infuseai/docker-stacks:base-notebook-1951c2b3-gpu-cuda-12	GPU CUDA 12	3.9