无法在多个GPU上运行tensorflow

时间:2016-01-18 17:07:34

标签: python gpu tensorflow

我正在从tensorflow存储库运行cifar10多GPU示例。我可以使用多个GPU。我的ubuntu PC有两个Titan X,我看到内存完全被两个GPU上的进程占用。但是,实际上只有一个GPU正在计算。我没有获得加速。我尝试过tensorflow 0.5.0和0.6.0 pip二进制文件。我也尝试过从源代码编译。

编辑: 安装旧版本的nvidia驱动程序后问题就消失了。

1 个答案:

答案 0 :(得分:3)

安装旧版本(352.55)的nvidia驱动程序后问题就消失了。