Tensorflow无法检测到第二个GPU

Question

在具有旧版GTX970的台式机上添加了新的GTX1050 ti。 Tensorflow仅检测到旧的GPU。尝试使用nvidia-smi，devicequery和pytorch，除了tensorflow以外，所有命令都找到了GPU。

系统配置：

• Ubuntu：18.04
• python 3.6
• Tensorflow-gpu：1.11.0
• Cuda：9.0
• Cudnn：7.0
• Nvidia驱动程序：384.87
• “ CUDA_VISIBLE_DEVICES” = 0,1

火炬测试命令：

1. torch.cuda.get_device_name（0）#nvidia：GTX 1050 ti
2. torch.cuda.get_device_name（1）#nvidia：GTX 970

均返回正确的GPU名称

Tensorflow测试命令：

1. print（device_lib.list_local_devices（））＃总是“ gpu：0
2. 使用tf.device（'/ device：GPU：1'）：#error：“ gpu：1”不存在

删除旧的，tensorflow可以找到新的。如果两个都安装，则仅检测到一个旧的。还尝试几次重新安装Cuda，Cudnn和tensorflow-gpu 1.11，仍然存在相同的错误。