配置NVIDIA容器运行时软件源:

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.repo | \
sudo tee /etc/yum.repos.d/nvidia-container-runtime.repo

安装NVIDIA容器运行时组件:

sudo yum install -y nvidia-container-runtime
sudo yum install -y nvidia-container-toolkit

配置Docker默认使用NVIDIA运行时:
方式一:

sudo vi /etc/docker/daemon.json

添加以下配置内容:

{
  "default-runtime": "nvidia",
  "runtimes": {
    "nvidia": {
      "path": "nvidia-container-runtime",
      "runtimeArgs": []
    }
  }
}
sudo systemctl daemaon-reload && systemctl restart docker

方式二(推荐):

配置 Docker 使用 NVIDIA 运行时

sudo nvidia-ctk runtime configure --runtime=docker

重启 Docker 服务

sudo systemctl restart docker

重启Docker服务应用配置:

sudo systemctl restart docker

启动Docker服务并验证GPU支持:

sudo systemctl start docker
docker run --gpus all ollama

使用了毫秒镜像加速器:
在这里插入图片描述

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐