Ollama Docker 镜像

本文内容均由Ollama官方文档翻译，仅供个人学习，如有差异请以官网文档为准（https://ollama.com）ollama.cadn.net.cn

仅限 CPU

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

NVIDIA GPU

安装 NVIDIA Container Toolkit。ollama.cadn.net.cn

使用 Apt 安装

1. 配置存储库ollama.cadn.net.cn

   curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey \
       | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
   curl -s -L 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 软件包ollama.cadn.net.cn

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

使用 Yum 或 Dnf 安装

1. 配置存储库ollama.cadn.net.cn

   curl -s -L 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 软件包ollama.cadn.net.cn

    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

[!NOTE]
如果您正在NVIDIA JetPack系统上运行，Ollama无法自动发现正确的JetPack版本。请向容器传递环境变量 JETSON_JETPACK=5 或 JETSON_JETPACK=6，以选择版本5或6。ollama.cadn.net.cn

AMD GPU

要使用 Docker 和 AMD GPU 运行 Ollama，请使用 rocm 标签并执行以下命令：ollama.cadn.net.cn

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

在本地运行模型

现在您可以运行模型了：ollama.cadn.net.cn

docker exec -it ollama ollama run llama3.2

尝试不同的模型

更多模型可在 Ollama 库 中找到。ollama.cadn.net.cn