python - Tensorflow 2.0 list_physical_devices 未检测到我的 GPU

Question

我最近在我的计算机上安装了 tensorflow 2.0，但是当我尝试在我的 GPU 上运行它时，函数 tf.config.experimental.list_physical_devices('GPU')在 Jupyter 或 Vitual Studio Code 上，它返回一个空数组。你知道为什么吗 ？

我的设置:

电脑:微星

处理器 : Intel(R) Core(TM) i7-8750H CPU @ 2.220GHz

GPU 0 : Intel(R) UHD Graphics 630

GPU : NVIDIA GeForce GTX 1060

Python:带有 Python 3.7 的 Ananconda 3

Tensenflow 2.0 已安装 pip install tensorflow
我的测试代码:

physical_devices = tf.config.experimental.list_physical_devices('GPU')
print(physical_devices)
if physical_devices:
  tf.config.experimental.set_memory_growth(physical_devices[0], True)

提前致谢 ! :)

Answer 1

在此处提供解决方案(答案部分)，即使它出现在评论部分是为了社区的利益。

而不是 pip install tensorflow ，你可以试试pip3 install --upgrade tensorflow-gpu或者只是删除 tensorflow然后 installing "tensorflow-gpu将解决您的问题。

安装 Tensorflow GPU 后，您可以检查 GPU 如下

physical_devices = tf.config.experimental.list_physical_devices('GPU')
print(physical_devices)
if physical_devices:
  tf.config.experimental.set_memory_growth(physical_devices[0], True)

输出:

[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]