Docker Images Reference
Official NVIDIA Images
CUDA Base Images
Image
Size
Use Case
# Recommended for most workloads
nvidia/cuda:12.8.0-base-ubuntu22.04PyTorch Images
Image
Size
PyTorch
CUDA
# Recommended for ML training
pytorch/pytorch:2.7.1-cuda12.8-cudnn9-develTensorFlow Images
Image
Size
TensorFlow
CUDA
Hugging Face Images
Image
Size
Use Case
Inference Servers
Image
Use Case
Development Environments
Image
Description
Using Custom Images
Build and Push
Use with Clore
Example Dockerfile
Image Selection Guide
For Training
Workload
Recommended Image
For Inference
Workload
Recommended Image
For Development
Workload
Recommended Image
Tips
See Also
Last updated
Was this helpful?