docker 支持 gpu

需求:

  • 原先在宿主机里运行的服务需要迁移到docker 里

进程:

  • docker 支持 gpu 需要装toolkit ,但是正常情况下没有对应的源,所以先引入源文件
bash 复制代码
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
&& curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
&& curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | \
sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
  • 导入并安装
bash 复制代码
apt-get update && apt-get install -y nvidia-container-toolkit
  • 启动docker 查看是否有--gpus 的命令
bash 复制代码
docker run --gpus all -it test:1.0 nvidia-smi
  • 有正常输出就没问题了

拓展:

  • docker 启动也可以指定显卡
bash 复制代码
docker run --gpus "device=1" -it bash nvidia-smi
相关推荐
眠修4 分钟前
Kuberrnetes 服务发布
linux·运维·服务器
hunjinYang22 分钟前
Tomcat镜像实战:掌握Dockerfile的编写以及发布项目
docker·tomcat
好奇的菜鸟1 小时前
Docker 配置项详解与示例
运维·docker·容器
xcs194051 小时前
集运维 麒麟桌面版v10 sp1 2403 aarch64 离线java开发环境自动化安装
运维·自动化
BAOYUCompany1 小时前
暴雨服务器成功中标华中科技大学集成电路学院服务器采购项目
运维·服务器
超龄超能程序猿2 小时前
Bitvisse SSH Client 安装配置文档
运维·ssh·github
奈斯ing2 小时前
【Redis篇】数据库架构演进中Redis缓存的技术必然性—高并发场景下穿透、击穿、雪崩的体系化解决方案
运维·redis·缓存·数据库架构
Connie14512 小时前
k8s多集群管理中的联邦和舰队如何理解?
云原生·容器·kubernetes
鳄鱼皮坡3 小时前
仿muduo库One Thread One Loop式主从Reactor模型实现高并发服务器
运维·服务器
即将头秃的程序媛3 小时前
centos 7.9安装tomcat,并实现开机自启
linux·运维·centos