在 Keras 中未检测到 GCP GPU

Posted 2023-02-23

【中文标题】在 Keras 中未检测到 GCP GPU

【英文标题】：GCP GPU is not detected in Keras

【发布时间】：2020-04-15 18:16:21

【问题描述】：

我正在使用一个 NVIDIA Tesla P4GPU 的 GCP 实例上运行 UNet Keras 模型。但它没有检测到 GPU。相反，它在 CPU 上运行。 p.s.我安装了驱动程序和 tensorflow-gpu 购买它不会工作。如何解决这个问题？

I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:156] kernel driver does not appear to be running on this host (instance-1): /proc/driver/nvidia/version does not exist
Num GPUs Available:  0

【问题讨论】：

错误信息包含你的答案，nvidia内核驱动加载了吗？

另外你需要一个支持 GPU 的 Tensorflow 版本。 Refer to this

如何找到 nvidia 内核驱动程序？

【参考方案1】：

您需要先安装驱动程序。 Follow this instruction