无法将NVIDIA Quadro P620 GPU与Tensorflow一起使用

Question

我已经安装了所有需要运行Tensorflow的GPU的东西，比如NVIDIA驱动程序，微软VisualStudio 2017 C++分发，CUDNN在正确的文件夹中。但我仍然无法使用GPU。我收到以下消息

---

以下是我的软件和硬件规格

• 我的GPU版本是NVIDIA Quadro P620
• Tensorflow和TF-GPU版本为2.3.0
• Keras和Keras GPU 2.4.3
• CUDA版本11.2.2
• NVIDIA驱动程序版本：27.21.14.6192

在哪里可以开始调试以解决此问题？有线索吗

Answer 1

根据Tensorflow tested build configuration，您需要为Tensorflow 2.3版本安装CUDA 10.1

如果Tensorflow、CUDA和cuDNN的版本之间存在任何不兼容，它将不会检测GPU

在您的情况下，我建议您将tensorflow版本升级到2.4 or 2.5，它支持CUDA 11.x

从tensorflow 2.0开始keras与Tensorflow集成，因此不需要单独的Keras or Keras-GPU安装。您可以使用tf.keras模块