Tensorflow无法检测到第二个GPU

在桌面上增加了一个新的GTX1050 ti,它有一个旧的GTX970 . Tensorflow仅检测到旧GPU . 尝试了nvidia-smi,devicequery和pytorch,除了tensorflow之外,所有命令都发现了gpus .

系统配置:

  • 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

火炬测试命令:

  • torch.cuda.get_device_name(0)#nvidia:GTX 1050 ti

  • torch.cuda.get_device_name(1)#nvidia:GTX 970

两者都返回正确的gpu名称

Tensorflow测试命令:

  • print(device_lib.list_local_devices())#always“gpu:0

  • with tf.device('/device:GPU:1'):#error:"gpu:1"不存在

删除旧的,tensorflow可以找到新的 . 如果两者都已安装,则仅检测到旧的 . 还尝试了多次重新安装Cuda,Cudnn和tensorflow-gpu 1.11,同样的错误仍然存在 .

回答(0)