社区所有版块导航
Python
python开源   Django   Python   DjangoApp   pycharm  
DATA
docker   Elasticsearch  
aigc
aigc   chatgpt  
WEB开发
linux   MongoDB   Redis   DATABASE   NGINX   其他Web框架   web工具   zookeeper   tornado   NoSql   Bootstrap   js   peewee   Git   bottle   IE   MQ   Jquery  
机器学习
机器学习算法  
Python88.com
反馈   公告   社区推广  
产品
短视频  
印度
印度  
Py学习  »  docker

怎样在 Docker 里调用 NVIDIA 显卡?适用于绿联,飞牛系统!

NASBox • 1 年前 • 1264 次点击  
如有修改或改动,关注文章底部留言!
教程对你有用,可以 “点赞” 和 “打赏”支持 ~



 

 

 

 

 

 

 

 

 

 

 

 

 

该篇教程主要讲解飞牛和绿联的 Docker 怎么才能调用独显,目前实测绿联有效,至于其他系统或者 AMD 显卡没办法测试

TIP:主要参考的是下面这位大佬的教程

https://zhuanlan.zhihu.com/p/8001362266

NVIDIA 驱动

NVIDIA 驱动是连接 NVIDIA 显卡硬件与操作系统、应用程序的关键软件,它的核心作用是让操作系统和应用能够识别、控制并高效利用 NVIDIA 显卡的硬件功能。

TIP:

目前我知道的绿联,飞牛和 Unriad 都是比较方便安装 NVIDIA 驱动的,至于其他 Linux 系统就不清楚了

绿联

飞牛

Unraid

输入下面命令测试是否成功

nvidia-smi

没问题的话会打印显卡信息

NVIDIA Container Toolkit 工具

NVIDIA Container Toolkit 是一套用于在容器化环境中高效运行 GPU 加速应用的工具集,它解决了容器与 NVIDIA GPU 之间的兼容性问题,让开发者和用户能轻松地在 Docker 等容器平台中利用 GPU 算力。

TIP:

下面代码适用于基于 Debian 系统的飞牛和绿联,至于其他系统可以参考下面官方文档:

https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html

切换 root 权限

sudo -i

执行 curl 命令导入仓库 GPG 密钥

curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/debian12/x86_64/3bf863cc.pub | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg

添加 NVIDIA Container Toolkit 官方仓库到系统源列表

echo "deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/debian12/x86_64 /" | sudo tee /etc/apt/sources.


    
list.d/nvidia-container-toolkit.list

更新包列表

sudo apt-get update

安装 nvidia-container-toolkit 工具包

sudo apt-get install -y nvidia-container-toolkit

配置 NVIDIA 在 Docker 中运行

sudo nvidia-ctk runtime configure --runtime=docker

重启 Docker 服务

sudo systemctl restart docker

运行 Docker 容器测试

docker run --rm --gpus all nvidia/cuda:11.3.1-base-ubuntu18.04 nvidia-smi

Docker 可以成功调用的话,就会显示显卡信息

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 


 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 


 

 

 

 

 

 


 

 

 

 

 

 


 

 

Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/184515