安裝 tensorflow-gpu 版本后,需要安裝相應的 CUDA 和 cuDNN
注意版本問題:tensorflow-gpu 1.7以及之后的版本要安裝 CUDA 8.0 以上的版本,tf 1.7 之下的安裝 CUDA 8.0 之下的 ,安裝 cuDNN 時要根據 CUDA 的版本來選擇
運行時錯誤如下:
根據錯誤信息安裝 CUDA 9.0 即可,注意是 9.0 不是9.2版本
CUDA 9.0 下載地址:https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exenetwork 下載完成后安裝即可
cuDNN 7.0.5 下載地址:https://developer.nvidia.com/rdp/cudnn-archive
1.下載 cuDNN 時需要注冊英偉達的賬號,建議使用 gmail 郵箱注冊,本人使用了QQ 和網易兩個郵箱都沒能收到驗證郵件,激活不了賬號,
2. cuDNN 下載完成后解壓,將解壓后的三個文件夾 bin,include,lib 復制到 CUDA 的安裝路徑下和重名的三個文件夾進行合並, CUDA 的默認安裝路徑為:
1 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
復制到CUDA 路徑 v9.0 文件夾下進行合並:
更多問題可參考:https://github.com/tensorflow/tensorflow/issues/17101