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バージョンを検出できません。 バージョン5または6を選択するには、環境変数JETSON_JETPACK=5またはJETSON_JETPACK=6をコンテナに渡してください。

AMD GPU

AMD GPUでDockerを使用して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

Vulkanを無効にするにはOLLAMA_VULKAN=0を使用し、特定のVulkanデバイスを選択するにはGGML_VK_VISIBLE_DEVICES=<ids>を使用します。

ローカルでモデルを実行

これでモデルを実行できます：

docker exec -it ollama ollama run llama3.2

他のモデルを試す

その他のモデルはOllamaライブラリで見つけることができます。