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。