
引言如果在 docker 中使用 gpu 时,遇到以下错误:
代码语言:bash复制```bash docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].
你可以按照以下步骤安装 nvidia-container-toolkit 来支持 docker 使用显卡。官方网站地址:
Installing the NVIDIA Container Toolkit — NVIDIA Container Toolkit 1.16.2 documentation
步骤使用 Yum 或 Dnf 进行安装配置生产仓库:
代码语言:bash复制
bash curl -s -L zuojiankuohaophpcnhttps:youjiankuohaophpcn | \sudo tee /etc/yum.repos.d/nvidia-container-toolkit.repozuojiankuohaophpcn/https:youjiankuohaophpcn
可选地,配置仓库以使用实验性包:
代码语言:bash复制bash sudo yum-config-manager --enable nvidia-container-toolkit-experimental
安装 NVIDIA Container Toolkit 包:代码语言:bash复制```bash
更新仓库sudo dnf clean allsudo dnf makecache# 安装 NVIDIA Container Toolkitsudo dnf install -y nvidia-container-toolkit
使用 nvidia-ctk 命令配置容器运行时:
代码语言:bash复制```bash
sudo nvidia-ctk runtime configure --runtime=docker
重启 Docker 守护进程:代码语言:bash复制```bash sudo systemctl restart docker
结论引用docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]]. AFTER installing nvidia-docker2 - Stack Overflow










