大家好

目前在 Win10 安裝 TensorFlow-gpu & Keras + CUDA
通常是沒什麼問題
例如參考

Win10 安裝 TensorFlow-gpu & Keras
https://reurl.cc/arzDEQ

 

這邊說明就相當詳細

主要就是下載 CUDA
https://developer.nvidia.com/

 

下載 cuDNN
https://developer.nvidia.com/cudnn

 

加入系統環境變數 %PATH%


這邊要注意，執行的介面如果打 echo %PATH% 無效 則該介面不能執行
例如我在最原始的 cmd 介面可以
但是在 power shell 就不行

安裝

pip install tensorflow-gpu

接著執行 tensorflow 要注意錯誤訊息
如果出現 Could not load dynamic library cudart64_101.dll
那個 101 有可能是 100 也有可能是 102
那就是要安裝對應版本的 CUDA 才對
例如 cudart64_101.dll 對應 CUDA 10.1

如果成功就會有以下訊息:

2020-06-03 10:18:35.385773: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll

大家一起快樂地使用 GPU 跑 tensorflow吧~