安装 nvidia-container-toolkit 以支持 docker 使用显卡
原创安装 nvidia-container-toolkit 以支持 docker 使用显卡
原创
叶子Tenney
发布于 2024-11-01 15:42:41
发布于 2024-11-01 15:42:41
文章被收录于专栏：叶子的数据科技专栏
引言
如果在 docker 使用 gpu 的时候，出现以下错误：
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
过程
Installing with Yum or Dnf
Configure the production repository:
curl -s -L <https://nvidia.github.io/libnvidia-container/stable/rpm/nvidia-container-toolkit.repo> | \
sudo tee /etc/yum.repos.d/nvidia-container-toolkit.repoOptionally, configure the repository to use experimental packages:
sudo yum-config-manager --enable nvidia-container-toolkit-experimentalInstall the NVIDIA Container Toolkit packages:
# 更新仓库
sudo dnf clean all
sudo dnf makecache
# 安装 NVIDIA Container Toolkit
sudo dnf install -y nvidia-container-toolkitConfigure the container runtime by using the nvidia-ctk command:
sudo nvidia-ctk runtime configure --runtime=dockerRestart the Docker daemon:
sudo systemctl restart docker结论
引用
原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。
如有侵权，请联系 cloudcommunity@tencent.com 删除。
原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。
如有侵权，请联系 cloudcommunity@tencent.com 删除。
评论
登录后参与评论
推荐阅读
目录
相关产品与服务
容器服务
腾讯云容器服务（Tencent Kubernetes Engine, TKE）基于原生 kubernetes 提供以容器为核心的、高度可扩展的高性能容器管理服务，覆盖 Serverless、边缘计算、分布式云等多种业务部署场景，业内首创单个集群兼容多种计算节点的容器资源管理模式。同时产品作为云原生 Finops 领先布道者，主导开源项目Crane，全面助力客户实现资源优化、成本控制。
腾讯云开发者
