安装 nvidia-container-toolkit 以支持 docker 使用显卡

引言如果在 docker 中使用 gpu 时，遇到以下错误：

代码语言：bash复制```bash docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].

<code>
你可以按照以下步骤安装 nvidia-container-toolkit 来支持 docker 使用显卡。
<p>官方网站地址：</p><p>Installing the NVIDIA Container Toolkit — NVIDIA Container Toolkit 1.16.2 documentation</p><p>步骤使用 Yum 或 Dnf 进行安装配置生产仓库：</p><p>代码语言：bash<svg fill="none" height="16" viewbox="0 0 16 16" width="16" xmlns="<a href="https://www.php.cn/link/c9041cfd2a40932691855abd98fd219a">http://www.w3.org/2000/svg"><path</a> clip-rule="evenodd" d="M4.5 15.5V3.5H14.5V15.5H4.5ZM12.5 5.5H6.5V13.5H12.5V5.5ZM9.5 2.5H3.5V12.5H1.5V0.5H11.5V2.5H9.5Z" fill="currentcolor" fill-rule="evenodd"></path></svg>复制<code>bash curl -s -L <https:> | \sudo tee /etc/yum.repos.d/nvidia-container-toolkit.repo</https:></code>

可选地，配置仓库以使用实验性包：

ColorMagic

AI调色板生成工具

下载

代码语言：bash复制bash sudo yum-config-manager --enable nvidia-container-toolkit-experimental

<code>
安装 NVIDIA Container Toolkit 包：</p><p>代码语言：bash<svg fill="none" height="16" viewbox="0 0 16 16" width="16" xmlns="<a href="https://www.php.cn/link/c9041cfd2a40932691855abd98fd219a">http://www.w3.org/2000/svg"><path</a> clip-rule="evenodd" d="M4.5 15.5V3.5H14.5V15.5H4.5ZM12.5 5.5H6.5V13.5H12.5V5.5ZM9.5 2.5H3.5V12.5H1.5V0.5H11.5V2.5H9.5Z" fill="currentcolor" fill-rule="evenodd"></path></svg>复制```bash</p><h1>更新仓库sudo dnf clean allsudo dnf makecache# 安装 NVIDIA Container Toolkitsudo dnf install -y nvidia-container-toolkit</code>

使用 nvidia-ctk 命令配置容器运行时：

代码语言：bash复制```bash

sudo nvidia-ctk runtime configure --runtime=docker

<code>
重启 Docker 守护进程：</p><p>代码语言：bash<svg fill="none" height="16" viewbox="0 0 16 16" width="16" xmlns="<a href="https://www.php.cn/link/c9041cfd2a40932691855abd98fd219a">http://www.w3.org/2000/svg"><path</a> clip-rule="evenodd" d="M4.5 15.5V3.5H14.5V15.5H4.5ZM12.5 5.5H6.5V13.5H12.5V5.5ZM9.5 2.5H3.5V12.5H1.5V0.5H11.5V2.5H9.5Z" fill="currentcolor" fill-rule="evenodd"></path></svg>复制```bash
sudo systemctl restart docker</code>

结论引用docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]]. AFTER installing nvidia-docker2 - Stack Overflow