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

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

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

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

 

更多详情请扫描下方二维码关注博主微信公众号:程序员大管