我正在 Ubuntu 18.04 上使用平台(Cytomine)来运行一些深度学习容器化应用程序(该平台自动处理 Docker 映像和容器,因此我只需要创建映像并向平台提供其下载 URL)。到目前为止,它运行良好,但现在我需要启用 GPU 支持才能有效运行模型。因此,我使用nvidia-docker进行了一些本地测试,以手动运行具有 GPU 支持的模型容器,让它工作非常容易,因为我只需向运行命令添加一个选项:
docker run --gpus all
但是,因为我无法将此选项添加到Cytomine平台上的代码中,所以我需要找到一种默认情况下向 docker 运行的所有容器添加/启用该选项的方法。
我尝试将此选项添加到文件中/etc/docker/daemon.json and /etc/docker/key.json
,然后重新启动 docker sudo systemctl restart docker
。然而,这并没有奏效。
另外,我还找到了如何创建 docker 配置文件(docker config);然而,这似乎只适用于 Docker Swarm,我不会在这个项目中使用 Swarm。
因此,我正在寻找一种可以正确部署的简单解决方案。在运行任何 Docker 容器时,有什么方法可以默认启用此选项(--gpus all)?(比如以某种方式将其包含在 Dockerfile 中?)
谢谢!
归档时间: |
|
查看次数: |
531 次 |
最近记录: |