注:此博客為不需要安裝cuda和cudnn。(直接搭建tensorflow-gpu環(huán)境)
注:當(dāng)前顯卡GTX1050烤蜕。(當(dāng)前文章寫(xiě)作測(cè)試日期2020年4月22日)
一兵拢、anaconda安裝
- 官方下載
- 配置conda鏡像源(主要一下主要安裝用它)
- 配置pip鏡像源
臨時(shí)使用: https://mirror.tuna.tsinghua.edu.cn/help/pypi/
配置(注意看評(píng)論區(qū)): https://developer.aliyun.com/mirror/pypi?spm=a2c6h.13651102.0.0.3e221b114TOXjb
二已慢、配置查看
-
打開(kāi)NVIDIA控制面板查看當(dāng)前電腦支持的CUDA。
image.png -
查看源中tensorflow-gpu版本
image.png - 查看gpu對(duì)應(yīng)的其他依賴包版本
查看官網(wǎng)鏈接:https://www.tensorflow.org/install/source#common_installation_problems
- 新建tensorflow-gpu環(huán)境
因?yàn)橹安榭串?dāng)前版本的顯卡驅(qū)動(dòng)递惋,默認(rèn)支持cuda9.1柔滔。所以此時(shí)新建python=3.6的環(huán)境。
conda create -n tensorflow-gpu python=3.6
- 開(kāi)始安裝:
conda install tensorflow-gpu=1.12.0
# 由于更換過(guò)conda鏡像源萍虽,不用擔(dān)心速度睛廊。
# 不過(guò)在安裝cudatoolkit和cudnn時(shí)會(huì)斷線,多執(zhí)行幾次該命令就好了
由于我已經(jīng)安裝過(guò)了杉编,然后寫(xiě)的總結(jié)超全,所以我換一個(gè)沒(méi)有安裝的演示一下:
執(zhí)行安裝:
查看得知安裝依賴所需的cudatoolkit=9.0.1,低于我電腦支持的9.1非常不錯(cuò)。那就選擇y繼續(xù)邓馒。如果一次沒(méi)有安裝上,多執(zhí)行自己安裝命令嘶朱。conda可是換了源的。
后面還有測(cè)試GPU是否生效绒净,別著急见咒。
也順便總結(jié)可能出現(xiàn)的如第6錯(cuò)誤:
-
常見(jiàn)錯(cuò)誤:
- 創(chuàng)建環(huán)境的python版本不對(duì)
如下,我在base環(huán)境下是python=3.7挂疆,所以無(wú)法安裝tensorflow=1.12.0版本改览,它也報(bào)了相應(yīng)的錯(cuò)誤,需要更換為3.6(上面第3點(diǎn)*查看gpu對(duì)應(yīng)的其他依賴包版本中有提到)缤言。順便提了一下我電腦支持cuda==9.0.1也在截圖中有提示宝当。
image.png
- 創(chuàng)建環(huán)境的python版本不對(duì)
- 安裝沒(méi)有指定版本的tensorflow
當(dāng)執(zhí)行
conda install tensorflow-gpu
由于沒(méi)有指定版本,會(huì)安裝最新的cuda和cudnn胆萧,由于我顯卡驅(qū)動(dòng)提示我電腦支持的cuda是9.1庆揩,所以并沒(méi)有安裝最新俐东,也并未指導(dǎo)是否生效。但是查看下面博客得知安裝最新會(huì)不生效订晌,未可知虏辫。如果你的顯卡驅(qū)動(dòng)支持cuda10的話,當(dāng)然可以安裝了锈拨。我在我ubuntu18.04上也是這樣安裝的最新版砌庄,也成功點(diǎn)亮gpu。
參考:https://blog.csdn.net/zpzhangpeng123/article/details/89295387
二奕枢、安裝
測(cè)試安裝是否成功
-
出現(xiàn)報(bào)錯(cuò):
image.png 解決報(bào)錯(cuò):
參考1-源碼解決(注意查看評(píng)論):https://blog.csdn.net/bigdream123/article/details/99467316
參考2-降低numpy版本(使用conda):https://blog.csdn.net/kobe_academy/article/details/99706595
- 開(kāi)始解決:
- 再次測(cè)試是否gpu生效:
參考: https://blog.csdn.net/weixin_43522055/article/details/99617967
至此大功告成娄昆。開(kāi)始tensorflow學(xué)習(xí)。
備注
安裝好了之后缝彬,興致勃勃跑到tensorflow光網(wǎng)去學(xué)習(xí):
what f**k萌焰!官網(wǎng)怎么只有2.0 的教程了(tensorflow2.x+python3.7+cuda10),雖然很多github高贊項(xiàng)目都是支持tensorflow1.x的但是我想要跟著官網(wǎng)學(xué)習(xí)谷浅。也不知道我電腦支不支持當(dāng)前方式安裝扒俯,還是需要老方式安裝更新顯卡驅(qū)動(dòng)然后在安裝cuda10 cudnn?先試試吧壳贪。
- 創(chuàng)建python=3.7的環(huán)境
- 激活創(chuàng)建的環(huán)境后陵珍,直接不指定版本,安裝最新版违施。
conda install tensorflow-gpu
顯然如下圖互纯,依賴是cuda10的,不知道能不能生效磕蒲。
開(kāi)始安裝后留潦,由于網(wǎng)絡(luò)帶寬會(huì)中斷,多執(zhí)行幾次辣往,conda配置了源兔院,不會(huì)太久。
- 使用之前的驗(yàn)證方式還是出問(wèn)題了U鞠鳌坊萝!
查看原因,不支持高版本许起,除非更新驅(qū)動(dòng)十偶,就吃裝cuda
- 突然想起我裝的是ubuntu18.04雙系統(tǒng),后來(lái)就沒(méi)有搞win了园细,哈哈哈惦积,然后在ubuntu上去學(xué)習(xí)了