TensorFlow 未显示 GPU

Question

我测试了以下代码：

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

检查我的 PC 上的 GPU 数量，但我收到以下错误代码：

C:\Users\PRodr\Desktop>py test.py
2020-04-09 12:18:50.929587: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
Traceback (most recent call last):
  File "test.py", line 1, in <module>
    from tensorflow.python.client import device_lib
  File "C:\Users\PRodr\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\__init__.py", line 41, in <module>
    from tensorflow.python.tools import module_util as _module_util
  File "C:\Users\PRodr\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\__init__.py", line 84, in <module>
    from tensorflow.python import keras
  File "C:\Users\PRodr\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\keras\__init__.py", line 27, in <module>
    from tensorflow.python.keras import models
  File "C:\Users\PRodr\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\keras\models.py", line 24, in <module>
    from tensorflow.python.keras import metrics as metrics_module
  File "C:\Users\PRodr\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\keras\metrics.py", line 37, in <module>
    from tensorflow.python.keras.engine import base_layer
  File "C:\Users\PRodr\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\keras\engine\base_layer.py", line 51, in <module>
    from tensorflow.python.keras import initializers
  File "C:\Users\PRodr\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\keras\initializers\__init__.py", line 127, in <module>
    populate_deserializable_objects()
  File "C:\Users\PRodr\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow\python\keras\initializers\__init__.py", line 85, in populate_deserializable_objects
    generic_utils.populate_dict_with_module_objects(
AttributeError: module 'tensorflow.python.keras.utils.generic_utils' has no attribute 'populate_dict_with_module_objects'

我尝试重新安装 TensorFlow，但没有成功。

我安装了：

• cuda_10.2.89_win10_network
• cudnn-10.2-windows10-x64-v7.6.5.32
• TensorRT-7.0.0.11.Windows10.x86_64.cuda-10.2.cudnn7.6

Answer 1

检查 GPU 是否存在的正确/推荐方法是：

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

此外，您应该提供更多信息，以便为您指明正确的方向（Cuda + CuDNN + TF. Version）