安装 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></pre><p>可选地，配置仓库以使用实验性包：</p><p>代码语言：bash<svg fill="none" height="16" viewbox="0 0 16 16" width="16" xmlns="http://www.w3.org/2000/svg"><path 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
sudo yum-config-manager --enable nvidia-container-toolkit-experimental</p><pre class="brush:php;toolbar:false;"><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><div class="aritcle_card flexRow">
                                                        <div class="artcardd flexRow">
                                                                <a class="aritcle_card_img" href="/ai/2663" title="靠岸学术"><img
                                                                                src="https://img.php.cn/upload/ai_manual/001/503/042/69b3bf2c510bf638.jpeg" alt="靠岸学术"  onerror="this.onerror='';this.src='/static/lhimages/moren/morentu.png'" ></a>
                                                                <div class="aritcle_card_info flexColumn">
                                                                        <a href="/ai/2663" title="靠岸学术">靠岸学术</a>
                                                                        <p>一款集翻译，阅读，文献管理于一体的英文文献阅读器</p>
                                                                </div>
                                                                <a href="/ai/2663" title="靠岸学术" class="aritcle_card_btn flexRow flexcenter"><b></b><span>下载</span> </a>
                                                        </div>
                                                </div><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