NVIDIA于2016年开始设计NVIDIA-Docker已便于容器使用NVIDIA GPUs。 第一代nvidia-docker1.0实现了对docker client的封装,并在容器启动时,将必要的GPU device和libraries挂载到容器中
安装前满足条件
驱动要求
英伟达驱动程序
在开始之前,请确保您已经为您的 Linux 发行版安装了 NVIDIA 驱动程序。安装驱动程序的推荐方法是使用您的发行版的包管理器,但也可以使用其他安装程序机制(例如,通过.run从 NVIDIA Driver Downloads下载安装程序)。
有关使用包管理器从官方 CUDA 网络存储库安装驱动程序的说明,请按照本指南中的步骤操作。
平台要求
运行 NVIDIA Container Toolkit 的先决条件列表如下所述:
- GNU/Linux x86_64 内核版本 > 3.10
- Docker >= 19.03(推荐,但某些发行版可能包含较旧版本的 Docker。支持的最低版本为 1.12)
- 具有架构 >= Kepler 的 NVIDIA GPU(或计算能力 3.0)
- NVIDIA Linux 驱动程序>= 418.81.07(请注意,不支持较旧的驱动程序版本或分支。)
安装方法
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html
验证安装
docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi