之前做tf(下文用tf指代Tensorflow)的相關(guān)開(kāi)發(fā)都是基于Linux和Mac,最近發(fā)現(xiàn)tf已經(jīng)對(duì)win有了支持,于是在自己的win本上進(jìn)行了相關(guān)的配置嘗試离福,遇到了一些小坑垦藏,也參考了網(wǎng)上一些前輩留下的技術(shù)資料,就用這篇文章記錄一下颈畸。
tf的GPU版本比CPU版本要快出很多乌奇,所以我首選GPU版本。GPU版本需要電腦內(nèi)置NVIDIA的獨(dú)立顯卡眯娱。我的電腦配置為8G內(nèi)存礁苗、i5-7300hq的cpu和GTX-1050的GPU。操作系統(tǒng)為win10徙缴。
目前win平臺(tái)僅支持python3.5试伙,這要特別注意!我之前安裝的最新的python3.6版本于样,在執(zhí)行pip命令安裝tf的時(shí)候會(huì)報(bào)錯(cuò)疏叨!
1、安裝anaconda穿剖。這個(gè)可以去官網(wǎng)下載:https://www.continuum.io/downloads/
下載完畢后一步步點(diǎn)擊安裝即可蚤蔓。我安裝在了D盤,安裝完后把:D:\Users\你的用戶名\Anaconda3\Scripts路徑添加到環(huán)境變量里糊余。方便cmd下使用conda秀又。
2、由于上文提及我安裝了python3.6啄刹,所以在cmd下利用anaconda運(yùn)行下面命令創(chuàng)造一個(gè)python3.5的環(huán)境:
conda create --name tensorflow python=3.5
activate tensorflow
conda install jupyter
conda install scipy
pip install tensorflow-gpu
3涮坐、安裝NVIDIA顯卡的兩個(gè)驅(qū)動(dòng):
CUDA安裝:https://developer.nvidia.com/cuda-downloads
下載完是一個(gè)exe文件,一步步點(diǎn)擊下一步完成安裝誓军!
CuDNN安裝:https://developer.nvidia.com/cudnn
需要注冊(cè)才能下載袱讹,我解壓到了C盤:C:\cuda(把它添加到系統(tǒng)環(huán)境變量)。這里有個(gè)坑!添加環(huán)境變量的時(shí)候一定不要忘記添加C:\cuda\bin捷雕,因?yàn)檫@里面有cudnn64_5.dll這個(gè)文件椒丧。你要是不添加,會(huì)報(bào)錯(cuò)救巷!
4壶熏、然后運(yùn)行pip install --upgrade --ignore-installed tensorflow-gpu
5、配置Pycharm你的項(xiàng)目的Project Interpreter浦译,選擇上文中建的那個(gè)名為tensorflow的python3.5環(huán)境棒假。
6、輸入下面代碼:
import tensorflow as tf
hello = tf.constant("Hello!TensorFlow")
sess = tf.Session()
print(sess.run(hello))
運(yùn)行:
出現(xiàn)上面結(jié)果精盅,證明配置成功帽哑!