如何使用 nvidia docker 作为提供者来运行 Vagrant

Question

我是开发机器学习应用程序的团队的一员。

目前我们正在使用 Vagrant 和 Docker 提供商作为统一的开发环境。

我们希望在开发过程中利用计算机上的 GPU，我发现 Nvidia 发布了 nvidia-docker 来为简单的 docker 容器启用此功能。

如何使用 nvidia-docker 作为 Vagrant 的提供者？

如果不可能，有没有等效的解决方案？

---

对于我们来说，在部署的同一个 docker 镜像之上进行开发非常重要，因为我们依赖于多个交互的开源库，并且我们希望在一个地方管理它们

（部署时不会破坏依赖项）

Answer 1

docker 19.0.3（发行说明）在其

--gpus

命令中添加了

docker run

参数

我们可以通过 vagrant 的

create_args

docker 配置选项提供，docs:

create_args（字符串数组）- 容器启动时传递给 docker run 的附加参数。这可用于设置未通过 Vagrantfile 公开的参数。

因此添加此行后，GPU 将在 VM 中公开

config.vm.provider "docker" do |d|
    ...
    d.create_args = ["--gpus", "all"]
    ...
end