tensorflow-gpu 2.3 安装与 cuda 和 cudnn 无法检测 GPU？

Question

我有一个关于在 tensorflow 中使用 GPU 的问题 我正在使用 RTX2060

我正在关注本教程Solution to TensorFlow 2 not using GPU

我已经安装了

cudatoolkit = 10.2.89

cudnn = 7.6.5

tensorflow-gpu = 2.3.0

但是当我用

签入python时

print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

原来我的 GPU 编号是 0

我试图在整个互联网上找到解决方案，但仍然无法解决。

请帮忙，谢谢。

Answer 1

因此，该指南不足以使用 GPU 设置 tensorflow；例如，这里有一个随机指南（其中有很多），解释了一些缺少的组件。请注意处理 CUDA 驱动程序、适当的 GPU 驱动程序版本、特定 linux 发行版 (Ubuntu 20.04) 等的部分：

random example of how to set up tensorlfow for use with a GPU

我不知道这是否是一个好的指南，或者它是否有效，但我只是想指出您正在使用的指南中一些重要的缺失部分。

因此，您必须使用适当的 GPU 驱动程序和适当的 linux 发行版设置 CUDA 驱动程序（我注意到您根本没有提及 CUDA 驱动程序版本）（我从未在 Windows 上设置过 CUDA，所以我知道那）为了让 tensorflow 能够识别你的 GPU。