Лише CPU

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

Nvidia GPU

Встановіть 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 GPU

Щоб запустити Ollama за допомогою Docker з AMD GPU, використовуйте тег 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.