TensorFlow 未捕获 GPU

Question

我在 Windows 上的 Anaconda 虚拟环境中使用 pip install tensorflow 安装了 tensorflow。

我尝试测试是否启用了 GPU，然后输入

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

得到了

Num GPUs Available:  0

我的系统确实启用了 CUDA 和 CUDNN，因为我在安装 PyTorch GPU 版本时没有问题。如何为 TensorFlow 启用 GPU？

Answer 1

首先，使用卸载tensorflow，

pip uninstall tensorflow

安装tensorflow-gpu版本，

pip install tensorflow-gpu==2.2.0 

如果使用 pip 不起作用，您可以尝试使用 conda install command.

conda install -c anaconda tensorflow-gpu 

这将自动安装CUDA & cuDNN. 希望这能解决您的问题。

Answer 2

使用pip uninstall tensorflow删除cpu版本的tensorflow并安装tensorflow的gpu版本pip install tensorflow-gpu。

您也可以查看本教程link。

可以概括为以下几个步骤：

1. 卸载旧的 tensorflow
2. 安装 tensorflow-gpu pip install tensorflow-gpu
3. 安装 Nvidia 显卡和驱动程序（您可能已经拥有）
4. 下载并安装 CUDA
5. 下载并安装 cuDNN
6. 使用您的程序进行验证。