GPU未被TensorFlow捕获

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

Installtensorflow-gpuversion，

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，并安装gpu版本的tensorflow,pip install tensorflow-gpu。

你也可以点击这个教程链接。

可归纳为以下步骤:

1. 卸载旧的tensorflow
2. 安装tensorflow-gpu
3. 安装Nvidia显卡&驱动程序(你可能已经有了)
4. 下载,安装CUDA
5. 下载,安装cuDNN
6. 验证您的程序。