天天看點

windows下cuda8.0安裝最新tensorflow問題

目前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.

繼續閱讀