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

相关推荐
9命怪猫3 小时前
[K8S小白问题集] - Calico好在哪里?
网络·云原生·容器·kubernetes
齐潇宇3 小时前
k8s-Helm管理器
linux·运维·云原生·容器·kubernetes
Zhu7584 小时前
[软件部署]在k8s环境部署alist
云原生·容器·kubernetes
o_insist4 小时前
Docker 入门:从镜像、容器到项目部署
docker·自动化运维·devops
冷小鱼5 小时前
Dockerfile 编写与优化完全指南:从入门到生产级实践
docker·docker file
Cat_Rocky6 小时前
K8S-Helm简单学习分享
学习·容器·kubernetes
ziqi5228 小时前
Docker compose 和共享数据
运维·docker·容器
Zhu7589 小时前
[配置管理]k8s集群中airflow的端口转发
云原生·容器·kubernetes
泓博9 小时前
Macbook Docker Compose不识别
运维·docker·容器
susu10830189119 小时前
windows系统的WSL的Ubuntu安装docker
linux·ubuntu·docker