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.5.1 and CUDA 12.4.1, ready for PyTorch model development.
pytorch/pytorch:2.5.1-cuda12.4.1-cudnn8-runtime
Docker image with PyTorch 2.6.0 and CUDA 12.4.1, ready for PyTorch model development.
pytorch/pytorch:2.6.0-cuda12.4.1-cudnn8-runtime
Docker image with PyTorch 2.7.0 and CUDA 12.6.3, ready for PyTorch model development.
pytorch/pytorch:2.7.0-cuda12.6.3-cudnn8-runtime
Docker image running Stable Diffusion Web UI to immediately start generating stunning generative AI images.
primeintellect/stable-diffusion
Docker Image running Axolotl, the best library to start fine-tuning various AI models.
axolotlai/axolotl-cloud
Image pre-installed with prime-rl and verifiers to enable RL and RFT training at scale.
primeintellect/prime-rl-devel
Docker Image running the Bittensor CLI to immediately start mining or validating on the Bittensor network.
opentensorfdn/bittensor
Create your own custom template with your specific requirements and configurations.
Summary
Review your GPU selection and configuration details.