怎样在 Docker 里调用 NVIDIA 显卡？适用于绿联，飞牛系统！

Hello 大家好，我是 David

订阅我的频道 ，分享更多 NAS 教程和资讯 ~

该篇教程主要讲解飞牛和绿联的 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 可以成功调用的话，就会显示显卡信息

作者声明本文无利益相关，欢迎值友理性交流，和谐讨论～