實際上壤短,安裝windows環(huán)境下的TensorFlow并不是太麻煩呢蛤,主要問題在于第一次不知道版本兼容性問題祭衩。像我這種強迫癥安裝軟件總是想裝最新版的人,就容易掉入坑中钦讳,導致CUDA矿瘦,TensorFlow,python愿卒,cudnn版本的不兼容缚去。
接下來,進入正題琼开,安裝一共需要三個步驟易结。在這之前先說一下作者的環(huán)境。
系統(tǒng):WIN10
CUDA: 9.2
TensorFlow:1.11(發(fā)文時最新版本)
GPU:1060
PS:在這之前柜候,先說一下版本匹配方式搞动,因為TensorFlow為主體軟件,因此:
1渣刷,在 GitHub TensorFlow release里面找到最新發(fā)布的版本鹦肿。
2辅柴,記下文中畫圈的cuDNN版本箩溃,到Nvidia官網(wǎng)找到cuDNN相應版本高诺。我下載的是如上圖圈中所示的7.2。
∧氪邸3虱而,上圖可以看到對應的CUDA版本為9.2,同樣在Nvdia官網(wǎng)CUDA archive下找到對應版本下載即可开泽。
安裝步驟
一牡拇,安裝VS
(一定確保先于CUDA安裝)
由于之后安裝CUDA的時候會去找VS的安裝路徑生成相應的CUDA文件夾,因此務必確保VS已經(jīng)安裝成功后再裝CUDA穆律。
VS應該是13 15 17版本都可以惠呼,作者使用的是15,網(wǎng)上說稍微穩(wěn)定一些峦耘,17的話可能會遇到奇怪的問題剔蹋。(強迫癥忍住了)其中VS的組件只需要安裝C++相關即可。
二辅髓,安裝CUDA
這一步?jīng)]有什么特別的泣崩,下載好相應的CUDA版本,只需要無腦下一步即可洛口。但是要注意的是矫付,安裝好后應該把以下文件夾添加到系統(tǒng)path中。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\libnvvp
安裝完成之后測試安裝結(jié)果:
(2),利用VS2015編譯測試文件
至此瞻佛,可以在cmd中切換到路徑bin\win64\Release下脱篙,然后運行bandwidthTest和deviceQueryDrv,可以看到運行結(jié)果伤柄,不在贅述绊困。
三,添加cuDNN
將下載好的cuDNN解壓縮适刀,復制內(nèi)容到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2路徑下秤朗,直接覆蓋即可。三笔喉,安裝TensorFlow
作者主要采用在線安裝的方式取视。
(1),安裝anaconda硝皂。這里也不在贅述安裝方式,主要注意需要下載python3.6作谭。
(3),創(chuàng)建好環(huán)境后在Anaconda prompt中輸入activate Tensorflow激活環(huán)境。
整個安裝過程到此就結(jié)束了扇谣∶两荩可以打開spyder測試一下代碼。如果不報錯就證明沒問題了揍堕。
import tensorflow as tf
with tf.device('/cpu:0'):
a = tf.constant([1.0,2.0,3.0],shape=[3],name='a')
b = tf.constant([1.0,2.0,3.0],shape=[3],name='b')
with tf.device('/gpu:1'):
c = a+b
sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True,log_device_placement=True))
sess.run(tf.global_variables_initializer())
print(sess.run(c))