PrimeHub
v4.1
v4.1
  • Introduction
  • Installation
  • Tiers and Licenses
  • End-to-End Tutorial
    • 1 - MLOps Introduction and Scoping the Project
    • 2 - Train and Manage the Model
    • 3 - Compare, Register and Deploy the Model
    • 4 - Build the Web Application
    • 5 - Summary
  • User Guide
    • User Portal
    • Notebook
      • Notebook Tips
      • Advanced Settings
      • PrimeHub Notebook Extension
      • Submit Notebook as Job
    • Jobs
      • Job Artifacts
      • Tutorial
        • (Part1) MNIST classifier training
        • (Part2) MNIST classifier training
        • (Advanced) Use Job Submission to Tune Hyperparameters
        • (Advanced) Model Serving by Seldon
        • Job Artifacts Simple Usecase
    • Models
      • Manage and Deploy Model
      • Model Management Configuration
    • Deployments
      • Pre-packaged servers
        • TensorFlow server
        • PyTorch server
        • SKLearn server
        • Customize Pre-packaged Server
        • Run Pre-packaged Server Locally
      • Package from Language Wrapper
        • Model Image for Python
        • Model Image for R
        • Reusable Base Image
      • Prediction APIs
      • Model URI
      • Tutorial
        • Model by Pre-packaged Server
        • Model by Pre-packaged Server (PHFS)
        • Model by Image built from Language Wrapper
    • Shared Files
    • Datasets
    • Apps
      • Label Studio
      • MATLAB
      • MLflow
      • Streamlit
      • Tutorial
        • Create Your Own App
        • Create an MLflow server
        • Label Dataset by Label Studio
        • Code Server
    • Group Admin
      • Images
      • Settings
    • Generate an PrimeHub API Token
    • Python SDK
    • SSH Server Feature
      • VSCode SSH Notebook Remotely
      • Generate SSH Key Pair
      • Permission Denied
      • Connection Refused
    • Advanced Tutorial
      • Labeling the data
      • Notebook as a Job
      • Custom build the Seldon server
      • PrimeHub SDK/CLI Tools
  • Administrator Guide
    • Admin Portal
      • Create User
      • Create Group
      • Assign Group Admin
      • Create/Plan Instance Type
      • Add InfuseAI Image
      • Add Image
      • Build Image
      • Gitsync Secret for GitHub
      • Pull Secret for GitLab
    • System Settings
    • User Management
    • Group Management
    • Instance Type Management
      • NodeSelector
      • Toleration
    • Image Management
      • Custom Image Guideline
    • Volume Management
      • Upload Server
    • Secret Management
    • App Settings
    • Notebooks Admin
    • Usage Reports
  • Reference
    • Jupyter Images
      • repo2docker image
      • RStudio image
    • InfuseAI Images List
    • Roadmap
  • Developer Guide
    • GitHub
    • Design
      • PrimeHub File System (PHFS)
      • PrimeHub Store
      • Log Persistence
      • PrimeHub Apps
      • Admission
      • Notebook with kernel process
      • JupyterHub
      • Image Builder
      • Volume Upload
      • Job Scheduler
      • Job Submission
      • Job Monitoring
      • Install Helper
      • User Portal
      • Meta Chart
      • PrimeHub Usage
      • Job Artifact
      • PrimeHub Apps
    • Concept
      • Architecture
      • Data Model
      • CRDs
      • GraphQL
      • Persistence Storages
      • Persistence
      • Resources Quota
      • Privilege
    • Configuration
      • How to configure PrimeHub
      • Multiple Jupyter Notebook Kernels
      • Configure SSH Server
      • Configure Job Submission
      • Configure Custom Image Build
      • Configure Model Deployment
      • Setup Self-Signed Certificate for PrimeHub
      • Chart Configuration
      • Configure PrimeHub Store
    • Environment Variables
Powered by GitBook
On this page
  • TensorFlow
  • PyTorch
  • Base
  1. Reference

InfuseAI Images List

PreviousRStudio imageNextDesign

Last updated 1 month ago

InfuseAI provides and maintains certain images which can be used as base images on 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

infuseai/docker-stacks↗