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的信息,說明安裝成功。這裏安裝的時候最好查看一下自己的顯卡支持的
相關文章
相關標籤/搜索