当前位置:   article > 正文

本地centos7+docker+ollama+gpu部署_ollama docker gpu

ollama docker gpu

1、一台有 NVIDIA GPU 驱动的机器

2、Docker CE安装

  1. # 删除旧版本的 Docker(如果存在)
  2. sudo yum remove -y docker docker-common docker-selinux docker-engine
  3. # 安装必要的软件包:
  4. sudo yum install -y yum-utils device-mapper-persistent-data lvm2
  5. # 添加 Docker CE 存储库:
  6. sudo yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo
  7. # 安装 Docker CE
  8. sudo yum install -y docker-ce
  9. # 启动 Docker 服务
  10. sudo systemctl start docker
  11. # 设置 Docker 开机自启
  12. sudo systemctl enable docker

3、安装NVIDIA docker Container Toolkit

  1. # 添加 NVIDIA Container Toolkit 存储库密钥
  2. distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
  3. curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo
  4. # 安装 NVIDIA Container Toolkit
  5. sudo yum install -y nvidia-docker2

4、安装NVIDIA docker Container Toolkit

  1. #创建或编辑 Docker 配置文件 /etc/docker/daemon.json
  2. vi /etc/docker/daemon.json
  3. # 添加以下内容到文件中,保存:
  4. {
  5. "default-runtime": "nvidia",
  6. "runtimes": {
  7. "nvidia": {
  8. "path": "nvidia-container-runtime",
  9. "runtimeArgs": []
  10. }
  11. }
  12. }

5、重启 Docker 服务

sudo systemctl restart docker

6、安装ollama,用gpu,docker

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

7、使用案例

docker exec -it ollama ollama run llama3

参考来源:

https://hub.docker.com/r/ollama/ollama

GitHub - ollama/ollama: Get up and running with Llama 3, Mistral, Gemma, and other large language models.

centOS 快速安装和配置 NVIDIA docker Container Toolkit_nvidia container toolkit-CSDN博客

本文内容由网友自发贡献,转载请注明出处:https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/736334
推荐阅读
相关标签
  

闽ICP备14008679号