tensorflow - TensorFlow 未捕获 GPU

Question

我在 Windows 上的 Anaconda 虚拟环境中安装tensorflow使用。pip install 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？

标签： tensorflowanacondagputensorflow2.0

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. 希望这将解决您的问题。