nvidia docker, nvidia docker2, nvidia container toolkits区别

最新推荐文章于 2025-04-23 13:04:50 发布

coco_1998_2

最新推荐文章于 2025-04-23 13:04:50 发布

阅读量1.1k

点赞数 13

分类专栏： # CUDA # Docker 文章标签： docker 容器

本文链接：https://blog.youkuaiyun.com/coco_1998_2/article/details/144716632

版权

Docker 同时被 2 个专栏收录

13 篇文章

订阅专栏

CUDA

3 篇文章

订阅专栏

背景

在docker容器中用GPU时，查阅了网上许多教程，教程之间概念模糊不清，相互矛盾，过时的教程和新的教程混杂在一起。主要原因是Nvidia为docker容器的支持发生了好几代变更，api发生了不少变化。下面来总结一下各代支持发展历程。

省流版总结

凡是使用了命令nvidia docker或者在docker中引入了--runtime=nvidia参数的都是过时教程，最新方法只需要下载nvidia-container-toolkits，在docker中引入--gpus参数即可。

nvidia docker

nvidia docker是NVIDIA第一代支持docker容器内使用GPU资源的项目。运行时用nvidia-docker命令。

根据nvidia docker在github ( https://github.com/NVIDIA/nvidia-docker )上的描述，已经不再使用了。

The nvidia-docker wrapper script that was included in this repository is no longer included in the package and a configuration specific to the target container engine (e.g. Docker, Containerd, Cri-o, or Podman) is suggested instead.

nvidia docker2

nvidia docker2是NVIDIA第二代支持docker容器内使用GPU资源的项目。运行时用nvidia-docker命令，且需要指定参数--runtime=nvidia.

根据 github (https://github.com/NVIDIA/nvidia-docker#backward-compatibility) 的描述，一代和二代之间有如下兼容性。

Backward compatibility To help transitioning code from 1.0 to 2.0, a bash script is provided in /usr/bin/nvidia-docker for backward compatibility. It will automatically inject the --runtime=nvidia argument and convert NV_GPU to NVIDIA_VISIBLE_DEVICES.

也就是说，在二代中，既可以使用nvidia docker命令，这会自动引入参数--runtime=nvidia也可以使用docker命令，手动指定参数--runtime=nvidia

如帖子 (https://stackoverflow.com/questions/50462153/nvidia-docker-run-vs-docker-run-runtime-nvidia) 总结：

docker run --runtime=nvidia is only available since nvidia-docker v2.
Both commands are equivalent with nvidia-docker v2, the former is a script provided for backward compatibility with nvidia-docker v1.