CUDNN_STATUS_INTERNAL_ERROR 升级到 tensorflow-gpu 1.9 后

CUDNN_STATUS_INTERNAL_ERROR after upgarde to tensorflow-gpu 1.9

我在 Ubuntu 16.04 上使用 python 2.7 和 tensorflow-gpu 1.6,效果很好。

使用 pip 升级到 tensorflow-gpu 1.9 后,我收到以下错误消息:

could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

我有 CUDA 9.0 和 cuDNN 7.0.5。

我在这里错过了什么?

RuntimeError: CUDNN_STATUS_INTERNAL_ERROR中给出的解决方案对我也有效。简而言之,运行

sudo rm -rf ~/.nv

然后重启。