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

相关推荐
天一生水water1 小时前
ubuntu使用毫秒镜像方式安装docker mysql
ubuntu·阿里云·docker
Qayrup4 小时前
docker 搭建私有仓库,推送并拉取
运维·docker·容器
郭庆汝4 小时前
docker拉取英伟达官方cuda11.8镜像
docker·cuda11.8
天下不喵5 小时前
Ubuntu24.04安装Docker过程记录
docker
黑黍6 小时前
如何在k8s中配置并使用nvidia显卡
云原生·容器·kubernetes
冷血~多好7 小时前
使用docker部署elk,实现日志追踪
elk·docker·容器
会飞的小蛮猪7 小时前
SkyWalking运维之路(Java探针接入)
java·运维·经验分享·容器·skywalking
天一生水water8 小时前
docker-compose安装
运维·docker·容器
蓝象_8 小时前
docker安装配置mysql
mysql·docker·容器
Cxzzzzzzzzzz9 小时前
Kubernetes 架构
容器·架构·kubernetes