升级到tensorflow-gpu 1.9后CUDNN_STATUS_INTERNAL_ERROR

时间:2018-07-24 14:43:50

标签: tensorflow cudnn

我在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。

我在这里想念什么?

1 个答案:

答案 0 :(得分:0)

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

sudo rm -rf ~/.nv

然后重新启动。