Только CPU

docker run -d -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama

Графический процессор Nvidia

Установите NVIDIA Container Toolkit.

Установка через Apt

1. Настройте репозиторий

   curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey \
       | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
   curl -fsSL https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list \
       | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' \
       | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
   sudo apt-get update

2. Установите пакеты NVIDIA Container Toolkit

   sudo apt-get install -y nvidia-container-toolkit

Установка через Yum или Dnf

1. Настройте репозиторий

   curl -fsSL https://nvidia.github.io/libnvidia-container/stable/rpm/nvidia-container-toolkit.repo \
       | sudo tee /etc/yum.repos.d/nvidia-container-toolkit.repo

2. Установите пакеты NVIDIA Container Toolkit

   sudo yum install -y nvidia-container-toolkit

Настройка Docker для использования драйвера Nvidia

sudo nvidia-ctk runtime configure --runtime=docker
sudo systemctl restart docker

Запуск контейнера

docker run -d --gpus=all -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama

Примечание

Если вы работаете на системе NVIDIA JetPack, Ollama не может автоматически определить нужную версию JetPack. Передайте контейнеру переменную окружения JETSON_JETPACK=5 или JETSON_JETPACK=6, чтобы выбрать версию 5 или 6.

Графический процессор AMD

Для запуска Ollama через Docker с GPU AMD используйте тег rocm и следующую команду:

docker run -d --device /dev/kfd --device /dev/dri -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama:rocm

Поддержка Vulkan

Vulkan встроен в образ ollama/ollama и включен по умолчанию, если контейнер имеет доступ к устройствам GPU.

docker run -d --device /dev/kfd --device /dev/dri -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama

Используйте OLLAMA_VULKAN=0, чтобы отключить Vulkan, или GGML_VK_VISIBLE_DEVICES=<ids>, чтобы выбрать конкретные устройства Vulkan.

Локальный запуск модели

Теперь вы можете запустить модель:

docker exec -it ollama ollama run llama3.2

Попробуйте другие модели

Больше моделей можно найти в библиотеке Ollama.