Python 无法在多个GPU上运行tensorflow

Python 无法在多个GPU上运行tensorflow,python,gpu,tensorflow,Python,Gpu,Tensorflow,我正在运行tensorflow存储库中的cifar10多GPU示例。我能够使用多个GPU。我的ubuntu PC有两个Titan X,我看到内存被两个GPU上的进程完全占用。然而,实际上只有一个GPU在计算。我得不到加速。我尝试过tensorflow 0.5.0和0.6.0 pip二进制文件。我也尝试过从源代码编译 编辑: 在我安装了旧版本的nvidia驱动程序后,问题消失了 在我安装了旧版本(352.55)的nvidia驱动程序后,问题消失了。用于运行我的机器的cifar10多GPU示例。在我

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

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

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

用于运行我的机器的cifar10多GPU示例。在我安装了更新的nvidia驱动程序和更新版本的tensorflow后,它无法使用多个GPU。你怎么知道它没有使用第二个GPU?此外,您还可以尝试使用log_device_placement=True进行会话。@YaroslavBulatov我通过安装旧版本的nvidia驱动程序解决了这个问题。顺便说一句,使用nvidia smi检查GPU利用率