Выберите готовый шаблон для быстрого старта или используйте свой Docker-образ
Фреймворк глубокого обучения от Meta AI. Включает CUDA, cuDNN и все необходимые зависимости.
pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
Комплексная платформа для машинного обучения от Google.
tensorflow/tensorflow:2.15.0-gpu
Интерактивная среда разработки с предустановленными библиотеками для ML.
jupyter/tensorflow-notebook:latest
Готовый образ для генерации изображений с Automatic1111 WebUI.
sd-webui/stable-diffusion:latest
Высокопроизводительный движок инференса для LLM.
vllm/vllm:latest
Полный набор инструментов Transformers с оптимизацией для GPU.
huggingface/transformers-pytorch-gpu:latest
Простой способ запуска LLM локально. Поддержка Mistral, Llama, Code Llama и других.
ollama/ollama:latest
Фреймворк глубокого обучения от Meta AI. Включает CUDA, cuDNN и все необходимые зависимости.
pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtimeКомплексная платформа для машинного обучения от Google.
tensorflow/tensorflow:2.15.0-gpuИнтерактивная среда разработки с предустановленными библиотеками для ML.
jupyter/tensorflow-notebook:latestГотовый образ для генерации изображений с Automatic1111 WebUI.
sd-webui/stable-diffusion:latestОбраз для запуска и дообучения моделей LLaMA 2.
meta-llama/llama:2-70bВысокопроизводительный движок инференса для LLM.
vllm/vllm:latestПолный набор инструментов Transformers с оптимизацией для GPU.
huggingface/transformers-pytorch-gpu:latestФреймворк для построения и обучения моделей разговорного ИИ.
nvcr.io/nvidia/nemo:24.01Простой способ запуска LLM локально. Поддержка Mistral, Llama, Code Llama и других.
ollama/ollama:latestМодульный интерфейс для Stable Diffusion с поддержкой нод-системы.
comfyanonymous/comfyui:latestБазовый образ для развёртывания ML API с поддержкой GPU.
gpucloud/fastapi-cuda:latestЧистый Ubuntu 22.04 с NVIDIA драйверами и CUDA Toolkit.
nvidia/cuda:12.3.1-devel-ubuntu22.04