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

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


xuqian02041524,发表于:2018-4-19 22:57:20

不是特别确定你的情况,不过我估计你需要用到 nvidia-docker。安装好了以后可以用 tensorflow 的 GPU 容器


yunhai_luo,发表于 2018-4-20 02:18:18

楼上正解,如果你在容器里陆续安装 NVIDIA 驱动,CUDA,cudnn,tensorflow,这跟复制系统没啥区别了。


TF-boy,发表于 2018-4-20 08:19:46

nvidia-smi nvidia-detector 可以帮到你


neverchange,发表于 2018-7-3 18:26:42

nvdia-docker 这个可以,头一次知道,收藏一下。


ViolinSolo,发表于 2018-7-3 19:45:38