InfuseAI Images List

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

TensorFlow

Framework	Image	CPU / GPU	Python
TensorFlow 2.17	infuseai/docker-stacks:tensorflow2.17.0-py3.11-b4826add	CPU	3.11
TensorFlow 2.17	infuseai/docker-stacks:tensorflow2.17.0-cuda12.2-py3.11-9d284a03	GPU CUDA 12	3.11
TensorFlow 2.14.1	infuseai/docker-stacks:tensorflow2.14.1-py3.9-dc0e7e15	CPU	3.9
TensorFlow 2.14.1	infuseai/docker-stacks:tensorflow2.14.1-cuda11.8-py3.9-f471492b	GPU CUDA 11	3.9
TensorFlow 2.17	infuseai/docker-stacks:tensorflow2.17.0-rocm6.3.1-py3.10-ubuntu24.04-bf6804d0	GPU ROCm 6.3.1	3.10

---

PyTorch

Framework	Image	CPU / GPU	Python
PyTorch v2.4.0	iinfuseai/docker-stacks:pytorch2.4.0-py3.11-b4826add	CPU	3.11
PyTorch v2.4.0	infuseai/docker-stacks:pytorch2.4.0-cuda12.2-py3.11-9d284a03	GPU CUDA 12	3.11
PyTorch 2.6.0	infuseai/docker-stacks:pytorch2.6.0-rocm6.3.1-py3.10-ubuntu24.04-bf6804d0	GPU ROCm 6.3.1	3.10

---

Base

Name	Image	CPU / GPU	Python
Base notebook	infuseai/docker-stacks:base-py3.11-b4826add	CPU	3.11
Base notebook	infuseai/docker-stacks:base-cuda12.2-py3.11-9d284a03	GPU CUDA 12	3.11
Base notebook	infuseai/docker-stacks:base-rocm6.3.1-py3.10-ubuntu24.04-bf6804d0	GPU ROCm 6.3.1	3.10