天天看點

win10環境下tensorflow-gpu安裝

win10環境下tensorflow-gpu安裝
踩了很多坑,浪費了一下午時間,在此記錄一下。

pip install tensorflow-gpu

然後下載下傳安裝cuda8.0

https://developer.nvidia.com/cuda-downloads
win10環境下tensorflow-gpu安裝

注意:隻能用8.0

我用9.0下載下傳了好久,程式提示我裝8.0

然後:下載下傳cudnn

https://developer.nvidia.com/rdp/cudnn-download
win10環境下tensorflow-gpu安裝

要注冊才能下載下傳,這又是一個坑,標明cudnn6.0版本

其中,cudnn的安裝方法是将cudnn下面的三個檔案夾(bin,include, lib)複制到cuda 8.0檔案夾。

裝完以後運作,出現以下錯誤:

ImportError: No module named '_pywrap_tensorflow_internal'

ImportError: DLL load failed: The specified module could not be found.           

解決的辦法是把cudnn64_6.dll改名為cudnn64_5.dll。

然後

import tensorflow

沒有報錯就成功了。

繼續閱讀