InfuseAI Images List
Last updated
Last updated
InfuseAI provides and maintains certain images which can be used as base images on on Docker Hub; here, the available images are:
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 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 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