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
相关推荐
JuiceFS1 天前
从 MLPerf Storage v2.0 看 AI 训练中的存储性能与扩展能力
运维·后端
chen9451 天前
mysql 3节点mgr集群部署
运维·后端
LH_R1 天前
OneTerm开源堡垒机实战(三):功能扩展与效率提升
运维·后端·安全
dessler1 天前
Hadoop HDFS-高可用集群部署
linux·运维·hdfs
少妇的美梦2 天前
logstash教程
运维
容器魔方2 天前
Bloomberg 正式加入 Karmada 用户组!
云原生·容器·云计算
chen9452 天前
k8s集群部署vector日志采集器
运维
chen9452 天前
aws ec2部署harbor,使用s3存储
运维
muyun28002 天前
Docker 下部署 Elasticsearch 8 并集成 Kibana 和 IK 分词器
elasticsearch·docker·容器
東雪蓮☆2 天前
深入理解 LVS-DR 模式与 Keepalived 高可用集群
linux·运维·服务器·lvs