系统是 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 设置容器的默认启动参数来解决
这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。
V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。
V2EX is a community of developers, designers and creative people.