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

Last updated