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

相关推荐
木雷坞6 分钟前
Docker / K8s 镜像拉取失败排查:专属加速域名与白名单配置思路
docker·容器·kubernetes
lwx9148526 分钟前
离线安装k8s 1.22.12版本
云原生·容器·kubernetes
涛声依旧-底层原理研究所13 分钟前
Docker+K8s:云原生应用基石
docker·kubernetes
yn0013 分钟前
Docker 一键部署加密支付网关:从零开始完整教程
运维·docker·容器
灰灰老师15 分钟前
Docker部署Tomcat9
java·linux·docker·tomcat
IT策士16 分钟前
第14篇 Docker Compose 开发环境最佳实践:热重载与调试
运维·docker·容器
正在走向自律20 分钟前
架构进阶:从 Docker 环境变量到 Nacos 统一配置中心实战
docker·容器·架构
comedate29 分钟前
[WSL2] 解决 WSL2 中 Docker 部署的 SearXNG 重启后,localhost 不能用的问题
docker·wsl2·searxng
Hui Baby37 分钟前
K8S自定义API
容器·贪心算法·kubernetes
ai产品老杨42 分钟前
突破异构算力与多协议壁垒:基于 Docker+边缘计算的企业级 AI 视频管理平台架构解析
人工智能·docker·边缘计算