Select your GPU type
Customize your cluster for optimal performance and scalability
Cluster base image
Select a pre-configured cluster setup tailored to your specific needs, requiring no extra configurations and ready to integrate with your codebase immediately.
Base image running Ubuntu 22 and CUDA 12. Ideal for devs who prefer to customize their environment. Fastest spin up times.
base image
Docker image with PyTorch 2.2.2 and CUDA 12.1, ready for PyTorch model development.
pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
Docker image with PyTorch 2.4.0 and CUDA 12.4.1, ready for PyTorch model development.
pytorch/pytorch:2.4.0-cuda12.4.1-cudnn8-runtime
Docker image running Stable Diffusion Web UI to immediately start generating stunning generative AI images.
primeintellect/stable-diffusion
Docker image running Flux ComfyUI to immediately start generating state-of-the-art AI images.
primeintellect/flux
Docker Image running Axolotl, the best library to start fine-tuning various AI models.
winglian/axolotl
Docker Image running the Bittensor CLI to immediately start mining or validating on the Bittensor network.
opentensorfdn/bittensor
Join the Prime Intellect distributed low-communication (DiLoCo) training with this Docker image.
primeintellect/open_diloco
Deploy your personal API instance of Llama3.1 8B Instruct via the VLLM inference library.
primeintellect/vllm-llama-3-1-8b-instruct
Create your own custom template with your specific requirements and configurations.
Summary
Review your GPU selection and configuration details.