僅限 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

若要在 Docker 中使用 AMD GPU 執行 Ollama，請使用 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 模型庫 中找到。