Tensorflow的那些坑-lib\site-packages\tensorflow\contrib\rnn\python\ops_gru_ops.dll not found

今天在使用使用cpu版的tensorflow跑卷积神经网络的时候非常慢,整整一天一夜6000次迭代还没有跑完。于是我就想把cpu版本的tensorflow换成gpu版本的。首先CUDA和cudnn已经安装完成,CUDA安装的是8.0版本,cudnn安装的是5.1版本,这两个版本要对应上不然也会报错。安装完成后在命令行里可以输出CUDA的信息,说明安装成功。这里安装的时候最好查看一下自己的显卡支持的CUDA的版本,再按照CUDA的版本去寻找cudnn的版本。

安装完成之后,开始使用 pip install tensorflow-gpu来安装,安装成功后在anaconda里使用import tensorflow时却报错了,报错信息如下:site-packages\tensorflow\contrib\rnn\python\ops_gru_ops.dll not found。说\python\ops_gru_ops模块没有找到,上来就怀疑是不是python的版本问题。后来登录https://tensorflow.google.cn/install/source_windows

网站查看了一下版本号,果然版本有问题,截图如下:

Tensorflow的那些坑-lib\site-packages\tensorflow\contrib\rnn\python\ops_gru_ops.dll not found

查看自己安装的python版本为3.6版本,对应的 tensorflow版本为1.2.0,果断卸载之前的版本使用pip  install tensorflow-gpu==1.2.0重新安装。

安装过后完美解决问题,所以大姐再安装gpu版本的时候一定要先检查自己的显卡对应的CUDA的型号,然后再安装对应的cudnn,最后在查找tensorflow对应的型号,这样就不会出错了。如果都没问题还是报错,那就卸载干净,重新安装一次。