七千二百袋水泥
七千二百袋水泥
Published on 2025-09-21 / 1 Visits

在Docker中调用NVIDIA显卡:适用于绿联和飞牛NAS系统的完整配置指南与实战步骤

本教程将详细讲解如何在绿联(UGREEN)和飞牛(Feilong)NAS系统中配置Docker以调用NVIDIA独立显卡,提供从驱动安装到容器工具集部署的全流程操作指南。

本文内容基于实际测试,确认在绿联NAS系统中有效可用。受条件限制,未对其他NAS系统或AMD显卡环境进行验证,建议用户根据自身设备参考操作。

注:部分配置方法参考了以下技术文章
https://zhuanlan.zhihu.com/p/8001362266

NVIDIA显卡驱动的安装与配置

NVIDIA驱动是连接显卡硬件与操作系统、应用程序的核心组件,其作用是让系统识别显卡硬件并高效调度其计算资源。目前绿联、飞牛及Unraid系统均提供了简便的NVIDIA驱动安装方式,其他Linux发行版的兼容性请以官方文档为准。

绿联NAS系统中的驱动安装

Image

飞牛NAS系统中的驱动安装

Image

Unraid系统中的驱动安装

Image

安装完成后,可通过以下命令验证驱动是否正常工作:

nvidia-smi

若配置成功,终端将显示当前显卡的详细信息与运行状态。

Image

部署NVIDIA Container Toolkit工具集

NVIDIA Container Toolkit是一套专为容器化环境设计的工具包,用于解决Docker等平台与NVIDIA GPU之间的兼容性问题,使容器内应用能够直接调用显卡计算资源。

注:以下操作适用于基于Debian的飞牛和绿联系统,其他系统请参考NVIDIA官方文档:
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html

首先获取root权限:

sudo -i

添加NVIDIA官方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

配置APT软件源:

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
sudo apt-get install -y nvidia-container-toolkit

配置Docker运行时支持:

sudo nvidia-ctk runtime configure --runtime=docker

重启Docker服务使配置生效:

sudo systemctl restart docker

运行测试容器验证GPU调用功能:

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

若一切正常,命令执行后将输出与本地系统一致的显卡信息。