
pip install tensorflow-gpu
然後下載下傳安裝cuda8.0
https://developer.nvidia.com/cuda-downloads
注意:隻能用8.0
我用9.0下載下傳了好久,程式提示我裝8.0
然後:下載下傳cudnn
https://developer.nvidia.com/rdp/cudnn-download
要注冊才能下載下傳,這又是一個坑,標明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
沒有報錯就成功了。