发新帖

基于GPU的tf源码安装问题请教?

[复制链接]
231 4

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
有个问题困扰多时,因为我需要部署多台机器,所以考虑使用docker(并不是为了使用官方的tensorflow的docker镜像),这样部署完以后就可以直接复制即可,
是否可以基于docker,先运行一个容器,然后在容器里陆续安装NVIDIA驱动,CUDA,cudnn,tensorflow?
网上也没有查到类似的安装方法,请各位大神多多指教:

yunhai_luo已获得悬赏 10 金币+5 金币

最佳答案

不是特别确定你的情况,不过我估计你需要用到nvidia-docker。安装好了以后可以用tensorflow的GPU容器。
本楼点评(0) 收起

精彩评论4

yunhai_luo  TF豆豆  发表于 2018-4-20 02:18:18 | 显示全部楼层
不是特别确定你的情况,不过我估计你需要用到nvidia-docker。安装好了以后可以用tensorflow的GPU容器
本楼点评(0) 收起
TF-boy  TF豆豆  发表于 2018-4-20 08:19:46 | 显示全部楼层
楼上正解,如果你在容器里陆续安装NVIDIA驱动,CUDA,cudnn,tensorflow,这跟复制系统没啥区别了
本楼点评(0) 收起
neverchange  TF豆豆  发表于 2018-7-3 18:26:42 | 显示全部楼层
nvidia-smi nvidia-detector可以帮到你
本楼点评(0) 收起
ViolinSolo  TF豆豆  发表于 2018-7-3 19:45:38 | 显示全部楼层
nvdia-docker 这个可以,头一次知道,收藏一下
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

快速回复 返回顶部 返回列表