在 debian 中用 docker 版本的 tensorflow-gpu 失效问题

2020-05-18 23:58:07 +08:00
 herozzm

本来要安装 gpu 版本,安装显卡驱动并安装和 nvidia 容器工具包后,运行docker run --gpus all --rm nvidia/cuda nvidia-smi,可以看到正常结果,然后正常安装 tensorflow-gpu docker 容器(里面的 tengsorflow 可以正常运行),完成后 exit 退出 ssh,过一段时间再次链接再次运行发现docker run --gpus all --rm nvidia/cuda nvidia-smi会卡死,同时 tensorflow 容器里面提示无法找到显卡,而运行其他非 docker 的 gpus 命令正常,有人遇到吗?百思不得其解,我尝试重新重启并从安装显卡驱动开始又恢复正常,然后过段时间后又失效

3301 次点击
所在节点    TensorFlow
0 条回复

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

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

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

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

© 2021 V2EX