今天在使用使用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對應的型號，這樣就不會出錯了。如果都沒問題還是報錯，那就卸載乾淨，重新安裝一次。