目前tensorflow版本已經到了1.6,對應的cuda版本為9.0,cudnn是v7,如果你之前裝的是cuda8的版本,現在按照官網的
pip3 install --upgrade tensorflow-gpu方式安裝tensorflow會預設裝的1.6,然後import tensorflow會提示找不到cuda還是cudnn的檔案
解決辦法1 還是用原來的cuda和cudnn,安裝指定的tensorflow1.4版本(不知道1.5是否一樣)
pip3 install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.4.0-cp35-cp35m-win_amd64.whl
上面的cp35和cp35m是你的python版本(3.5)
解決方法2 下載下傳cuda9和cudnn v7,按照官網pip3 install --upgrade tensorflow-gpu安裝。
2018/3/20.