Windows10下tensorflow 1.8 CUDA GPU加速版本安裝幾個坑

張國平發表於2018-06-16
做策略,想跟跟風,加入人工神經網路,deep learning。就試試學習下最火的tensorflow吧

然後看到gpu加速版本效能比cpu好很多,就裝了試試,發現坑比較多。安裝還是比較麻煩,可以讓很多初學者提前放棄;
其實cpu版本還是簡單很多,老黃的玩意支援性果然欠缺。

首先是確認下你的顯示卡支援不,這時候就後悔沒有買個遊戲本;小米pro的mx 150顯示卡照實聊勝於無,NVIDIA官網顯示還是支援的。

第一個坑: cuda 9.0並不支援 mx150,提示這個硬體不支援,那就下載最新的cuda 9.2 吧,這個時候注意,windows自帶安全套件會把cuda 9.2當作毒軟,遮蔽防毒軟體先。對了cudnn解壓縮的要複製出去覆蓋。

第二個坑:現在pip 下載的版本的tensorflow 1.8 gpu不是對應cuda9.2打包的,第一次試執行時候,會提示cudart64_90.dll 找不到;這個時候發現,其實cuda路徑裡面已經是cudart64_92.dll 版本了。這個時候很無解。搞了半天,一個是去github下載1.9版本;或者用下面這個版本,用cuda 9.2 recompile的。
https://github.com/fo40225/tensorflow-windows-wheel/blob/master/1.8.0/py36/GPU/cuda92cudnn71sse2/tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl

來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/22259926/viewspace-2156269/,如需轉載,請註明出處,否則將追究法律責任。

相關文章