docker中调用GPU算力(debain12系统)

前提,你的debain12系统已经安装好cuda和cudnn以及NVIDIA驱动程序,并且测试过已经可以调用GPU显卡算力

一、安装nvidia-docker

1.添加 NVIDIA 官⽅ Docker APT 仓库的 GPG 密钥:

curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | sudo apt-key add -

2.添加 NVIDIA 官⽅ Docker APT 仓库:

distribution=$(. /etc/os-release;echo I D ID IDVERSION_ID)

curl -s -L https://nvidia.github.io/nvidia-container-runtime/debian10/nvidia-container-runtime.list\| sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list

3.更新 APT 仓库并安装 nvidia-docker2:

sudo apt-get update && sudo apt-get install nvidia-container-runtime

4.重启docker

sudo systemctl restart docker

5.运行命令

docker run -d --name experience

-p 8508:8508 -p 8509:8509

-v /root/api/panoramic_recognition_experience_api.py:/mmdetection/panoramic_recognition_experience_api.py

-v /root/api/periapical_recognition_experience_api.py:/mmdetection/periapical_recognition_experience_api.py

-v /root/api/config.py:/mmdetection/config.py

-v /root/api/Dental_model:/mmdetection/Dental_model

-v /root/api/experience:/mmdetection/experience

--gpus all experience

相关推荐
古城小栈4 分钟前
K8s 1.30 新特性:AI 驱动的资源调度 深度解析
人工智能·容器·kubernetes
杨浦老苏30 分钟前
隐私优先的图片遮蔽工具AutoRedact
docker·ai·工具·群晖·图片
Radan小哥2 小时前
Docker学习笔记—day012
笔记·学习·docker
qq_5470261792 小时前
Docker 搭建Nexus3私服
运维·docker·容器
古城小栈2 小时前
K3s + 边缘 AI:轻量级 K8s 在嵌入式设备的部署
人工智能·容器·kubernetes
小道士写程序3 小时前
Kubernetes 1.23.17 集群部署完全记录(单点)
云原生·容器·kubernetes
driver19993 小时前
hyperV装的windows11安装docker不支持虚拟化
运维·docker·容器
warrah3 小时前
docker portainer的应用
docker
一点事4 小时前
centos7:离线安装docker
运维·docker·容器
初学者_xuan5 小时前
K8S资源无法删除处理方法
容器·贪心算法·kubernetes