tensorflow-gpu需要先安裝CUDA和cudnn,,具體參考 安裝CUDA和cudnn需要注意版本問(wèn)題,,具體查看tensorflow官網(wǎng) https://tensorflow.google.cn/install/source_windows 安裝Anaconda 下載地址: https://www./products/individual 在安裝Anaconda的過(guò)程中和其它軟件沒(méi)有區(qū)別,但是要注意不要勾選**“Add Anaconda to my PATH enviroment variable”,,我們后面會(huì)手動(dòng)配置環(huán)境變量,。 接下來(lái)就是等待了,安裝結(jié)束后需要測(cè)試是否能正常使用,,打開(kāi)CMD輸入“conda”命令,,發(fā)現(xiàn)提示“’conda’ is not recognized as an internal or external command, operable program or batch file.”,這時(shí)我們需要手動(dòng)配置環(huán)境變量 將以下三個(gè)路徑加入,,注意這里要換成你自己的安裝路徑,。
然后點(diǎn)擊“確定”保存,這回再測(cè)試一下,,再cmd中輸入“conda -V”,,能正常顯示版本號(hào),證明已經(jīng)配置好了,。 在安裝好Anaconda和CUDA后可以開(kāi)始配置tensorflow-gpu+keras+jupyter環(huán)境 通過(guò)調(diào)用下列命令,,創(chuàng)建一個(gè)名為“tensorflowGPU”的conda環(huán)境 conda create -n tensorflowGPU pip python=3.6 創(chuàng)建環(huán)境后通過(guò)下面的指令激活環(huán)境 activate tensorflowGPU 創(chuàng)建好環(huán)境并激活進(jìn)入后就可以開(kāi)始安裝需要的包了,這里推薦使用pip清華源進(jìn)行加速下載
升級(jí)環(huán)境中的pip版本(為了避免某些包可能不能安裝) python -m pip install –upgrade pip 升級(jí)pip如果遇到安裝慢的問(wèn)題,,同樣可以使用清華加速下載 python -m pip install -i https://pypi.tuna./simple –upgrade pip 安裝tensorflow-gpu1.5.0
安裝完后進(jìn)入python環(huán)境輸入測(cè)試程序驗(yàn)證安裝 import tensorflow as tf hello = tf.constant('Hello, TensorFlow!’) sess = tf.Session() print(sess.run(hello)) 若沒(méi)有報(bào)錯(cuò)并顯示如下則安裝成功 若報(bào)錯(cuò)了如果是numpy報(bào)錯(cuò)可以嘗試更換更低版本的numpy以適應(yīng)tensorflow-gpu1.5.0 pip install -i https://pypi.tuna./simple numpy==1.15.4 安裝好tensorflow-gpu后可以使用以下命令安裝keras pip install -i https://pypi.tuna./simple keras==2.1.4 驗(yàn)證keras安裝,,輸入import keras,若沒(méi)報(bào)錯(cuò)并顯示如下則安裝成功 最后安裝jupyter notebook pip install -i https://pypi.tuna./simple ipython pip install -i https://pypi.tuna./simple jupyter 安裝當(dāng)前環(huán)境的內(nèi)核(關(guān)鍵) python –m ipykernel install —user 這個(gè)指令會(huì)在C盤(pán)安裝一個(gè)jupyter notebook內(nèi)核,,這個(gè)指令會(huì)指定jupyter notebook打開(kāi)后使用的環(huán)境,,如果你在其它環(huán)境中使用這個(gè)指令則會(huì)把jupyter notebook打開(kāi)后使用的環(huán)境切換到執(zhí)行這個(gè)指令的環(huán)境 驗(yàn)證jupyter notebook的安裝,和使用的環(huán)境 jupyter notebook進(jìn)入jupyter notebook import keras 若顯示的版本對(duì)應(yīng)你的環(huán)境的話表示配置成功 若想改變jupyter notebook的工作空間可以參考 https://www.jianshu.com/p/462e0b0f1ba3 驗(yàn)證是否能成功調(diào)用GPU,,若出現(xiàn)以下結(jié)果說(shuō)明成功 至此tensorflow-gpu+keras+jupyter notebook的環(huán)境搭建完成 在后續(xù)自己安裝包時(shí)可能會(huì)出現(xiàn)各種版本沖突問(wèn)題,,這里給出我的環(huán)境中各種包的版本以供參考 |
|
來(lái)自: 袁先森lemon > 《深度學(xué)習(xí)》