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

1. 下載安裝Anaconda

簡(jiǎn)單說(shuō)就是下載 64位 python 3.5 版本的Anaconda

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

安裝情況：新機(jī)，未裝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 時(shí)有errorNo module named "_pywrap_tensorflow"或者DLL load failed, checkMSVCP140.DLLis in%PATH%。解決方法：install theVisual C++ 2015 redistributable(x64 version). 不過(guò)我沒(méi)遇到這個(gè)情況。
（我的做法是按照ImportError: No module named ‘_pywrap_tensorflow’ (MSVCP140.dll is present) #7705提供的一種做法，就是將cudnn6.0換成cudnn5.1的版本）

c. Anaconda虛擬環(huán)境內(nèi)外只能存在一個(gè)tensorflow，所以多出來(lái)的要?jiǎng)h除。因?yàn)锳naconda searches system site-packages from.localwith higher priority.

2. 進(jìn)入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

安裝完后可以看到下面幾個(gè)包

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

安裝成功后查看版本

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

然后你會(huì)發(fā)現(xiàn)我們找到不到各種lib，不急，因?yàn)槲覀冞€沒(méi)裝呢。

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

3. 安裝cuDNN和CUDA Toolkit

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

3.1 下載 cuDNN:

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

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

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

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

如果6.0版本安裝有錯(cuò)，請(qǐng)嘗試5.1版本。感謝酒肉丸子的分享

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

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

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

(C:\Users\xxxx\AppData\Local\Temp\CUDA 這個(gè)臨時(shí)文件夾不用多慮，安裝完了它會(huì)自動(dòng)刪除)

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

理論上這個(gè)步驟會(huì)自動(dòng)把路徑安裝到你的windows path里。

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

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

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

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

把那三個(gè)文件放到相同文件名的文件夾中。

萬(wàn)事俱備只欠重啟！

重啟電腦后，再次進(jìn)入Anaconda Prompt

輸入python進(jìn)入

import tensorflow

看到幾個(gè) successfully opened

恭喜你GPU版安裝完畢！

注意: 如果導(dǎo)入tensorflow時(shí)依然報(bào)錯(cuò)，請(qǐng)到path環(huán)境變量下確認(rè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

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

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

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

總結(jié)

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

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