最新tensorflow用不了GPU，cuda对应的tensorflow版本

我的电脑已经安装了cuda和cudnn，也安装了tensorflow-gpu，但是跑的时候无法调用GPU运算

通过一下代码检测GPU是否被使用：

import os
from tensorflow.python.client import device_lib
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"
if __name__ == "__main__":
    print(device_lib.list_local_devices())

output

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}

但是我这里只有一个CPU被调用, GPU无法调用。
下面介绍解决方法

注意：最重要的问题，你的cuda版本要配置对应的tensorflow-gpu。

查看你的cuda对应tensorflow-gpu版本：https://tensorflow.google.cn/install/source_windows
我的cuda版本为11.0， 在命令行（CMD）中输入nvidia-smi查看。最终发现自己的tensorflow-gpu版本装错了，装的系统默认的最新版本2.6.0，而我应该装2.4.0，怪不得报这样的错。

装成功后的检测结果界面是这样的：

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 14584989243831407567
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 3127299278
locality {
  bus_id: 1
  links {
  }
}
incarnation: 9780177882687983642
physical_device_desc: "device: 0, name: GeForce GTX 1050 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1"

安利一下大家需要的版本：

Windows最新的版本如下：

Linus/Macos最新的版本如下：