Шаблоны Docker-образов

Шаблоны Docker-образов

Выберите готовый шаблон для быстрого старта или используйте свой Docker-образ

Популярные шаблоны

PyTorch

Фреймворки ML

Фреймворк глубокого обучения от Meta AI. Включает CUDA, cuDNN и все необходимые зависимости.

pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime

TensorFlow

Фреймворки ML

Комплексная платформа для машинного обучения от Google.

tensorflow/tensorflow:2.15.0-gpu

Jupyter Lab

IDE

Интерактивная среда разработки с предустановленными библиотеками для ML.

jupyter/tensorflow-notebook:latest

Stable Diffusion

Генерация изображений

Готовый образ для генерации изображений с Automatic1111 WebUI.

sd-webui/stable-diffusion:latest

vLLM

LLM

Высокопроизводительный движок инференса для LLM.

vllm/vllm:latest

Hugging Face

Фреймворки ML

Полный набор инструментов Transformers с оптимизацией для GPU.

huggingface/transformers-pytorch-gpu:latest

Ollama

LLM

Простой способ запуска LLM локально. Поддержка Mistral, Llama, Code Llama и других.

ollama/ollama:latest

Все шаблоны

PyTorch

Фреймворки ML
Популярный

Фреймворк глубокого обучения от Meta AI. Включает CUDA, cuDNN и все необходимые зависимости.

pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime

TensorFlow

Фреймворки ML
Популярный

Комплексная платформа для машинного обучения от Google.

tensorflow/tensorflow:2.15.0-gpu

Jupyter Lab

IDE
Популярный

Интерактивная среда разработки с предустановленными библиотеками для ML.

jupyter/tensorflow-notebook:latest

Stable Diffusion

Генерация изображений
Популярный

Готовый образ для генерации изображений с Automatic1111 WebUI.

sd-webui/stable-diffusion:latest

LLaMA 2

LLM

Образ для запуска и дообучения моделей LLaMA 2.

meta-llama/llama:2-70b

vLLM

LLM
Популярный

Высокопроизводительный движок инференса для LLM.

vllm/vllm:latest

Hugging Face

Фреймворки ML
Популярный

Полный набор инструментов Transformers с оптимизацией для GPU.

huggingface/transformers-pytorch-gpu:latest

NVIDIA NeMo

LLM

Фреймворк для построения и обучения моделей разговорного ИИ.

nvcr.io/nvidia/nemo:24.01

Ollama

LLM
Популярный

Простой способ запуска LLM локально. Поддержка Mistral, Llama, Code Llama и других.

ollama/ollama:latest

ComfyUI

Генерация изображений

Модульный интерфейс для Stable Diffusion с поддержкой нод-системы.

comfyanonymous/comfyui:latest

FastAPI + CUDA

Развёртывание

Базовый образ для развёртывания ML API с поддержкой GPU.

gpucloud/fastapi-cuda:latest

Ubuntu + CUDA

Базовый образ

Чистый Ubuntu 22.04 с NVIDIA драйверами и CUDA Toolkit.

nvidia/cuda:12.3.1-devel-ubuntu22.04