系统是 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 设置容器的默认启动参数来解决
1
zlandjj 2020-02-26 02:24:18 +08:00
From 之前的 image
重做构建一个 image, 重写 cmd 或者 entrypoint |
3
Trim21 2020-02-26 03:27:38 +08:00
|
4
nvkou 2020-02-26 03:57:23 +08:00
没看懂.
-it 模式下是创建一个伪端交互. 你运行的是 bash. 这个其实可以直接写好你的启动命令 如果是给容器的参数,在 docker run 里写.比如你容器指定的 gpu 如果是给容器里应用的参数,多半要使用 -e 指定 environment 实际情况要参考你镜像的说明 如果你只是懒得打启动命令,写一个 docker-compose 就行了 |
5
d0m2o08 2020-02-26 08:00:21 +08:00
docker-compose ? 参数都写到 yml 里
|
6
silverfox 2020-02-26 08:09:19 +08:00
可以参考这个 Issue 中所提到的替代方法 Docker Support --gpus command line option youtrack.jetbrains.com/issue/IDEA-219125
|
7
fdppzrl 2020-02-26 08:51:10 +08:00 via Android
alias ?
|