标签: python gpu tensorflow
我正在从tensorflow存储库运行cifar10多GPU示例。我可以使用多个GPU。我的ubuntu PC有两个Titan X,我看到内存完全被两个GPU上的进程占用。但是,实际上只有一个GPU正在计算。我没有获得加速。我尝试过tensorflow 0.5.0和0.6.0 pip二进制文件。我也尝试过从源代码编译。
编辑: 安装旧版本的nvidia驱动程序后问题就消失了。
答案 0 :(得分:3)
安装旧版本(352.55)的nvidia驱动程序后问题就消失了。