轉(zhuǎn)載自在Windows上安裝GPU版Tensorflow。

1. 下載安裝Anaconda

簡單說就是下載 64位 python 3.5 版本的Anaconda

https://www.continuum.io/downloads#windows

安裝情況：新機，未裝python。

注意

a. Windows只支持64位 python 3.5

https://www.tensorflow.org/versions/master/get_started/os_setup#pip_installation_on_windows

b. 如果import tensorflow as tf 時有errorNo module named "_pywrap_tensorflow"或者DLL load failed, checkMSVCP140.DLLis in%PATH%。解決方法：install theVisual C++ 2015 redistributable(x64 version). 不過我沒遇到這個情況。
（我的做法是按照ImportError: No module named ‘_pywrap_tensorflow’ (MSVCP140.dll is present) #7705提供的一種做法，就是將cudnn6.0換成cudnn5.1的版本）

c. Anaconda虛擬環(huán)境內(nèi)外只能存在一個tensorflow，所以多出來的要刪除。因為Anaconda searches system site-packages from.localwith higher priority.

2. 進入Anaconda 菜單欄下的“Anaconda Prompt”

2.1 更新pip

輸入命令 python -m pip install –upgrade pip

2.2 安裝Tensorflow

按照不同需求輸入相應(yīng)的 命令， 記得刪去 “C:>”

CPU-only version

pip install --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-0.12.1-cp35-cp35m-win_amd64.whl --ignore-installed

非GPU版在這里就可以結(jié)束了。在Anaconda Prompt輸入python 后就能 導(dǎo)入tensorflow 包了。

我們著重講講GPU版！

pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl --ignore-installed

安裝完后可以看到下面幾個包

Installing collected packages: appdirs, six, pyparsing, packaging, setuptools, protobuf, numpy, wheel, tensorflow-gpu

安裝成功后查看版本

python -c "import tensorflow; print(tensorflow.__version__)”

然后你會發(fā)現(xiàn)我們找到不到各種lib，不急，因為我們還沒裝呢。

（https://github.com/ContinuumIO/anaconda-issues/issues/542）!!!

3. 安裝cuDNN和CUDA Toolkit

(win + R 后輸入dxdiag可以看自己的顯卡配置，用來選對應(yīng)的英偉達開發(fā)包驅(qū)動。)

3.1 下載 cuDNN:

NVIDIA CUDA? Deep Neural Network library (cuDNN) 大概54mb

https://developer.nvidia.com/rdp/cudnn-download

下載文件不穩(wěn)定，多試幾次（我下了三次）

這是個壓縮文件包，解壓縮后是三個文件夾，各有一個文件在里面。

如果6.0版本安裝有錯，請嘗試5.1版本。感謝酒肉丸子的分享

3.2 下載CUDA Toolkit 8.0: （我是下的8.0）

文件大小約1.2 G，下載文件不穩(wěn)定，花了很久時間，重復(fù)用net 安裝版

https://developer.nvidia.com/cuda-downloads

(C:\Users\xxxx\AppData\Local\Temp\CUDA 這個臨時文件夾不用多慮，安裝完了它會自動刪除)

另外你要是裝了360 就要不斷“允許該進程所有操作 ”

理論上這個步驟會自動把路徑安裝到你的windows path里。

(這里，我安裝的情況是，在我第一次安裝CUDA的時候，會先卸載了系統(tǒng)的NVDIA控制面板，然后自動就重啟了，然后其實還需要再次雙擊下載的cuda的exe文件，然后這次才會成功安裝。)

3.3 把cuDNN的文件復(fù)制到CUDAToolkit 安裝目錄

應(yīng)該在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 （根據(jù)你的選擇可能會有所不同）

總之你可以在那個目錄下找到一堆文件夾，其中包括了那三個解壓縮后的三個文件。

把那三個文件放到相同文件名的文件夾中。

萬事俱備只欠重啟！

重啟電腦后，再次進入Anaconda Prompt

輸入python進入

import tensorflow

看到幾個 successfully opened

恭喜你GPU版安裝完畢！

注意: 如果導(dǎo)入tensorflow時依然報錯，請到path環(huán)境變量下確認路徑

To use cuDNN with TensorFlow, the filecudnn64_5.dll must be in your%PATH%environment variable.

http://stackoverflow.com/questions/41007279/tensorflow-on-windows-couldnt-open-cuda-library-cudnn64-5-dll

一些其他的特殊情況：

https://github.com/tensorflow/tensorflow/issues/5968

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin

寫完發(fā)現(xiàn)了兩篇也不錯的指導(dǎo)

http://www.itdadao.com/articles/c15a832274p0.html

https://zhuanlan.zhihu.com/p/24369784?refer=wjdml

總結(jié)

以上是生活随笔為你收集整理的(转）在Windows上安装GPU版Tensorflow的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。