docker 支持 gpu

需求:

  • 原先在宿主机里运行的服务需要迁移到docker 里

进程:

  • docker 支持 gpu 需要装toolkit ,但是正常情况下没有对应的源,所以先引入源文件
bash 复制代码
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
&& curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
&& curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | \
sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
  • 导入并安装
bash 复制代码
apt-get update && apt-get install -y nvidia-container-toolkit
  • 启动docker 查看是否有--gpus 的命令
bash 复制代码
docker run --gpus all -it test:1.0 nvidia-smi
  • 有正常输出就没问题了

拓展:

  • docker 启动也可以指定显卡
bash 复制代码
docker run --gpus "device=1" -it bash nvidia-smi
相关推荐
saynaihe4 小时前
ubuntu 22.04 anaconda comfyui安装
linux·运维·服务器·ubuntu
企鹅与蟒蛇4 小时前
Ubuntu-25.04 Wayland桌面环境安装Anaconda3之后无法启动anaconda-navigator问题解决
linux·运维·python·ubuntu·anaconda
小蜜蜂爱编程4 小时前
ubuntu透网方案
运维·服务器·ubuntu
AI视觉网奇5 小时前
git 访问 github
运维·开发语言·docker
G_whang5 小时前
jenkins使用Jenkinsfile部署springboot+docker项目
spring boot·docker·jenkins
头发那是一根不剩了5 小时前
nginx:SSL_CTX_use_PrivateKey failed
运维·服务器
七夜zippoe6 小时前
破解 VMware 迁移难题:跨平台迁移常见问题及自动化解决方案
运维·自动化·vmware
hweiyu007 小时前
docker简介
运维·docker·容器
Sally璐璐7 小时前
OpenVPN:深度解析开源 VPN 解决方案
运维·开源
阿巴~阿巴~8 小时前
理解Linux文件系统:从物理存储到统一接口
linux·运维·服务器