如何设置 docker 容器的默认启动参数

2020-02-26 00:41:18 +08:00
 drperry

系统是 Ubuntu 18.03LTS,由于最新版 docker 已经原生支持 Nvidia GPU,只需要在启动时加上--gpus 参数即可,不用安装 nvidia docker,所以每次启动 tensorflow 镜像时都必须手动指定--gpus device=0 [以前的--runtime nvidia 参数已经废弃了]

就类似这样:
docker run --gpus device=0 -it tensorflow/tensorflow:2.0.1-gpu-py3 bash

但问题是现在的 Pycharm 并不支持自定义 docker 容器启动参数(只支持-v 等参数,不支持--gpus),就不能够在容器里调用 gpu,所以想通过给 docker 设置容器的默认启动参数来解决

4530 次点击
所在节点    程序员
7 条回复
zlandjj
2020-02-26 02:24:18 +08:00
From 之前的 image
重做构建一个 image, 重写 cmd 或者 entrypoint
Trim21
2020-02-26 03:24:47 +08:00
@zlandjj #1 这个是 docker run 命令的参数,不是容器内应用的参数。。。
Trim21
2020-02-26 03:27:38 +08:00
nvkou
2020-02-26 03:57:23 +08:00
没看懂.
-it 模式下是创建一个伪端交互. 你运行的是 bash. 这个其实可以直接写好你的启动命令
如果是给容器的参数,在 docker run 里写.比如你容器指定的 gpu
如果是给容器里应用的参数,多半要使用 -e 指定 environment 实际情况要参考你镜像的说明

如果你只是懒得打启动命令,写一个 docker-compose 就行了
d0m2o08
2020-02-26 08:00:21 +08:00
docker-compose ? 参数都写到 yml 里
silverfox
2020-02-26 08:09:19 +08:00
可以参考这个 Issue 中所提到的替代方法 Docker Support --gpus command line option youtrack.jetbrains.com/issue/IDEA-219125
fdppzrl
2020-02-26 08:51:10 +08:00
alias ?

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/647571

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX