在Keras中未检测到GCP GPU

时间:2019-12-24 18:59:22

标签: tensorflow keras deep-learning google-compute-engine nvidia

我正在带有一个NVIDIA Tesla P4GPU的GCP实例上运行UNet Keras模型。但是它无法检测到GPU。而是在CPU上运行。 ps。我安装了驱动程序和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

1 个答案:

答案 0 :(得分:0)

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