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