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

相关推荐
eventer1239 小时前
构建 HertzBeat Docker 镜像的技术实践
运维·docker·容器
Wokoo711 小时前
Docker镜像与分层深入理解
docker·云原生·容器
Empty_77712 小时前
K8S-Job & Cronjob
java·linux·docker·容器·kubernetes
Leon_Chenl12 小时前
使用 Docker 搭建全志 SDK 编译构建环境
docker·容器·全志 sdk·嵌入式 linux 构建编译·sdk 构建编译
zfj32113 小时前
Docker和容器OCI规范的关系
java·docker·eureka
qq_2130594313 小时前
从零搭建企业级K8s集群
云原生·容器·kubernetes
NewBee_Lxx14 小时前
docker nginx
运维·docker·容器
乌云下的风15 小时前
Doker 学习 小册
docker
龙亘川16 小时前
【课程2.4】开发环境搭建:K8s集群部署、芋道框架集成、ThingsBoard对接
java·容器·kubernetes·智慧城市·智慧城市一网统管 ai 平台
AI云原生17 小时前
openEuler软件生态体验——从桌面到服务器的全方位探索
运维·服务器·windows·macos·docker·云计算·腾讯云