阅读背景:

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

来源:互联网 

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




你的当前访问异常,请进行认证后继续阅读剩余内容。

分享到: