Windows 2012 編譯安裝Tensorflow

零揍鸟、環(huán)境準(zhǔn)備

git下載:https://git-for-windows.github.io/

MINGW下載:http://www.mingw.org/

Tensorflow源碼下載:git clone https://github.com/tensorflow/tensorflow/tree/r1.3

cmake for windows:https://cmake.org/download/

swig(python for c++ interface):http://www.swig.org/download.html

Visual Studio 2015:

Anaconda 4.2.0(此版本非最新版本,之所以使用是因?yàn)槠淠J(rèn)python版本為3.5.2):https://www.anaconda.com/download/

Numpy:https://www.scipy.org/scipylib/download.html

一、安裝CUDA

0. 首先需要在cuda-gpus檢查電腦的GPU是否支持cuda編程,要求Compute Capability 3.0及以上爷恳。

1. 安裝Visual Studio 2015(需要在CUDA之前安裝好)

2. 安裝CUDA toolkit 8.0(推薦下載離線版本)

3. 解壓下載好的cuDNN,將其中的 bin耕陷、include、lib 覆蓋到CUDA的安裝目錄

二然想、安裝tensorflow-gpu

1. 安裝anaconda

2. 進(jìn)入python3.5環(huán)境,通過以下命令安裝tensorflow-gpu:

pip install --ignore-installed --upgrade tensorflow-gpu

3. 如果遇到各種lib欣范、dll缺失的問題变泄,可以嘗試通過添加對應(yīng)的路徑到path中來解決

在CUDA_PATH后面添加bin和lib\x64路徑

將cudnn64_7改為cudnn64_6

4. 測試代碼

如下:

#Creates a graph.

a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')

b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')

c = tf.matmul(a, b)

#Creates a session with log_device_placement set to True.

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

#Runs the op.

print sess.run(c)

三、編譯tensorflow-gpu

官方只提供了Ubuntu和Mac OS X的編譯支持恼琼,在Windows下可以通過Bazel和CMake兩種方式進(jìn)行編譯妨蛹,但只是 “highly experimental”,可能會(huì)遇到各種錯(cuò)誤晴竞。下面使用CMake來進(jìn)行編譯蛙卤。

0. 進(jìn)行編譯之前安裝一下軟件:

Cmake 3.5以上

Git

swig

Windows下的額外要求:

Visual Studio 2015

Python 3.5

Numpy 1.11.0 or later

1. 設(shè)置環(huán)境變量

進(jìn)入目錄 C:\Program Files (x86)\Microsoft Visual Studio version\VC\,運(yùn)行命令 vcvarsall amd64颓鲜,從64位命令行編譯到64位的目標(biāo)平臺(tái)表窘。執(zhí)行這一步可以避免編譯的時(shí)候出現(xiàn)內(nèi)存不足的錯(cuò)誤。

將CUDA dlls和cuDNN dll添加到環(huán)境變量:

NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin

cudnn-8.0-windows7-x64-v7\cuda\bin

將Cmake和git添加到環(huán)境變量

CMake\bin

2. 下載 Tensorflow源代碼, 在其中的 tensorflow\tensorflow\contrib\cmake 目錄下新建 build 目錄甜滨,用于存放編譯結(jié)果乐严。

D:\temp> git clone https://github.com/tensorflow/tensorflow.git

D:\temp> cd tensorflow\tensorflow\contrib\cmake

D:\temp\tensorflow\tensorflow\contrib\cmake> mkdir build

D:\temp\tensorflow\tensorflow\contrib\cmake> cd build

D:\temp\tensorflow\tensorflow\contrib\cmake\build>

3. 調(diào)用cmake來產(chǎn)生Visual Studio solution 和project files.

D:\...\build> cmake .. -A x64 -DCMAKE_BUILD_TYPE=Release ^

More? -DSWIG_EXECUTABLE=C:/tools/swigwin-3.0.10/swig.exe ^

More? -DPYTHON_EXECUTABLE=C:/Users/%USERNAME%/AppData/Local/Continuum/Anaconda3/python.exe ^

More? -DPYTHON_LIBRARIES=C:/Users/%USERNAME%/AppData/Local/Continuum/Anaconda3/libs/python35.lib

如果有NVIDIA顯卡,且安裝好了cuDNN,那么可以添加以下參數(shù):

More? -Dtensorflow_ENABLE_GPU=ON ^

More? -DCUDNN_HOME="D:\...\cudnn"

4. 調(diào)用MSBiuld來編譯Tensorflow

使用以下命令產(chǎn)生.whl文件

D:\...\build> MSBuild /p:Configuration=Release tf_python_build_pip_package.vcxproj

遇到的問題:

error C2001: newline in constant

打開對應(yīng)的文件發(fā)現(xiàn)出現(xiàn)了亂碼衣摩,google之后發(fā)現(xiàn)可以通過修改locale為 英語(美國) 來解決昂验。

編譯過程中下載失敗或者 git clone緩慢

使用命令 nslookup 查找失效ip(github.global.ssl.fastly.net,storage.googleapis.com等)捂敌,修改host文件。完了之后 ipconfig /flushdns既琴。當(dāng)然也可以嘗試手動(dòng)下載到對應(yīng)的目錄占婉。

參考guide:

https://github.com/tensorflow/tensorflow/tree/r1.3/tensorflow/contrib/cmake

http://blog.csdn.net/DawnRanger/article/details/77755412

http://blog.csdn.net/xingwei_09/article/details/72822360

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市甫恩,隨后出現(xiàn)的幾起案子逆济,更是在濱河造成了極大的恐慌,老刑警劉巖磺箕,帶你破解...
    沈念sama閱讀 206,723評論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件奖慌,死亡現(xiàn)場離奇詭異,居然都是意外死亡松靡,警方通過查閱死者的電腦和手機(jī)简僧,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,485評論 2 382
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來雕欺,“玉大人岛马,你說我怎么就攤上這事⊥懒校” “怎么了啦逆?”我有些...
    開封第一講書人閱讀 152,998評論 0 344
  • 文/不壞的土叔 我叫張陵,是天一觀的道長脸哀。 經(jīng)常有香客問我蹦浦,道長扭吁,這世上最難降的妖魔是什么撞蜂? 我笑而不...
    開封第一講書人閱讀 55,323評論 1 279
  • 正文 為了忘掉前任,我火速辦了婚禮侥袜,結(jié)果婚禮上蝌诡,老公的妹妹穿的比我還像新娘。我一直安慰自己枫吧,他們只是感情好浦旱,可當(dāng)我...
    茶點(diǎn)故事閱讀 64,355評論 5 374
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著九杂,像睡著了一般颁湖。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上例隆,一...
    開封第一講書人閱讀 49,079評論 1 285
  • 那天甥捺,我揣著相機(jī)與錄音,去河邊找鬼镀层。 笑死镰禾,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播吴侦,決...
    沈念sama閱讀 38,389評論 3 400
  • 文/蒼蘭香墨 我猛地睜開眼屋休,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了备韧?” 一聲冷哼從身側(cè)響起劫樟,我...
    開封第一講書人閱讀 37,019評論 0 259
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎织堂,沒想到半個(gè)月后毅哗,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 43,519評論 1 300
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡捧挺,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 35,971評論 2 325
  • 正文 我和宋清朗相戀三年虑绵,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片闽烙。...
    茶點(diǎn)故事閱讀 38,100評論 1 333
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡翅睛,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出黑竞,到底是詐尸還是另有隱情捕发,我是刑警寧澤,帶...
    沈念sama閱讀 33,738評論 4 324
  • 正文 年R本政府宣布很魂,位于F島的核電站扎酷,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏遏匆。R本人自食惡果不足惜法挨,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,293評論 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望幅聘。 院中可真熱鬧凡纳,春花似錦、人聲如沸帝蒿。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,289評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽葛超。三九已至暴氏,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間绣张,已是汗流浹背答渔。 一陣腳步聲響...
    開封第一講書人閱讀 31,517評論 1 262
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留胖替,地道東北人研儒。 一個(gè)月前我還...
    沈念sama閱讀 45,547評論 2 354
  • 正文 我出身青樓豫缨,卻偏偏與公主長得像,于是被迫代替她去往敵國和親端朵。 傳聞我的和親對象是個(gè)殘疾皇子好芭,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 42,834評論 2 345

推薦閱讀更多精彩內(nèi)容