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

Question

我在 tensorflow 中使用 GPU 时遇到问题 我正在使用 RTX2060

我正在从本教程中关注不使用 GPU 的 TensorFlow 2 解决方案

我已经安装

cudatoolkit = 10.2.89

cudnn = 7.6.5

张量流 GPU = 2.3.0

但是当我签入python时

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

原来我的 GPU 编号是 0

我已经尝试在整个互联网上找到解决方案，但仍然无法修复它。

请帮忙，谢谢。

标签： pythontensorflow

Answer 1

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

如何设置 tensorlfow 以与 GPU 一起使用的随机示例

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

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