Tensorflow无法在GPU上启动会话

Question

我正在使用GPU在Conda上运行Tensorflow。每次我收到此错误

E tensorflow/stream_executor/cuda/cuda_dnn.cc:353] Could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED
2018-10-09 11:58:43.850743: E tensorflow/stream_executor/cuda/cuda_dnn.cc:361] Possibly insufficient driver version: 390.48.0

我正在使用Ubuntu 18.04 Lambda

似乎是驱动程序问题。我该如何解决？

Answer 1

  

驱动程序版本可能不足：390.48.0

听起来像您需要升级Nvidia驱动程序。这样做的方式取决于您的发行版以及一开始如何安装驱动程序。

您还可以尝试降级Tensorflow安装。