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