注:此博客為不需要安裝cuda和cudnn。(直接搭建tensorflow-gpu環(huán)境)
注:當(dāng)前顯卡GTX1050。(當(dāng)前文章寫作測試日期2020年4月22日)
一、anaconda安裝
- 官方下載
- 配置conda鏡像源(主要一下主要安裝用它)
- 配置pip鏡像源
臨時使用: https://mirror.tuna.tsinghua.edu.cn/help/pypi/
配置(注意看評論區(qū)): https://developer.aliyun.com/mirror/pypi?spm=a2c6h.13651102.0.0.3e221b114TOXjb
二、配置查看
-
打開NVIDIA控制面板查看當(dāng)前電腦支持的CUDA。
image.png -
查看源中tensorflow-gpu版本
image.png - 查看gpu對應(yīng)的其他依賴包版本
查看官網(wǎng)鏈接:https://www.tensorflow.org/install/source#common_installation_problems

- 新建tensorflow-gpu環(huán)境
因為之前查看當(dāng)前版本的顯卡驅(qū)動,默認支持cuda9.1。所以此時新建python=3.6的環(huán)境。
conda create -n tensorflow-gpu python=3.6
- 開始安裝:
conda install tensorflow-gpu=1.12.0
# 由于更換過conda鏡像源,不用擔(dān)心速度。
# 不過在安裝cudatoolkit和cudnn時會斷線,多執(zhí)行幾次該命令就好了

由于我已經(jīng)安裝過了,然后寫的總結(jié),所以我換一個沒有安裝的演示一下:

執(zhí)行安裝:

查看得知安裝依賴所需的cudatoolkit=9.0.1,低于我電腦支持的9.1非常不錯。那就選擇y繼續(xù)。如果一次沒有安裝上,多執(zhí)行自己安裝命令。conda可是換了源的。
后面還有測試GPU是否生效,別著急。
也順便總結(jié)可能出現(xiàn)的如第6錯誤:
-
常見錯誤:
- 創(chuàng)建環(huán)境的python版本不對
如下,我在base環(huán)境下是python=3.7,所以無法安裝tensorflow=1.12.0版本,它也報了相應(yīng)的錯誤,需要更換為3.6(上面第3點*查看gpu對應(yīng)的其他依賴包版本中有提到)。順便提了一下我電腦支持cuda==9.0.1也在截圖中有提示。
image.png
- 創(chuàng)建環(huán)境的python版本不對
- 安裝沒有指定版本的tensorflow
當(dāng)執(zhí)行
conda install tensorflow-gpu
由于沒有指定版本,會安裝最新的cuda和cudnn,由于我顯卡驅(qū)動提示我電腦支持的cuda是9.1,所以并沒有安裝最新,也并未指導(dǎo)是否生效。但是查看下面博客得知安裝最新會不生效,未可知。如果你的顯卡驅(qū)動支持cuda10的話,當(dāng)然可以安裝了。我在我ubuntu18.04上也是這樣安裝的最新版,也成功點亮gpu。
參考:https://blog.csdn.net/zpzhangpeng123/article/details/89295387

二、安裝
測試安裝是否成功
-
出現(xiàn)報錯:
image.png 解決報錯:
參考1-源碼解決(注意查看評論):https://blog.csdn.net/bigdream123/article/details/99467316
參考2-降低numpy版本(使用conda):https://blog.csdn.net/kobe_academy/article/details/99706595
- 開始解決:

- 再次測試是否gpu生效:
參考: https://blog.csdn.net/weixin_43522055/article/details/99617967

至此大功告成。開始tensorflow學(xué)習(xí)。
備注
安裝好了之后,興致勃勃跑到tensorflow光網(wǎng)去學(xué)習(xí):
what f**k!官網(wǎng)怎么只有2.0 的教程了(tensorflow2.x+python3.7+cuda10),雖然很多github高贊項目都是支持tensorflow1.x的但是我想要跟著官網(wǎng)學(xué)習(xí)。也不知道我電腦支不支持當(dāng)前方式安裝,還是需要老方式安裝更新顯卡驅(qū)動然后在安裝cuda10 cudnn?先試試吧。
- 創(chuàng)建python=3.7的環(huán)境

- 激活創(chuàng)建的環(huán)境后,直接不指定版本,安裝最新版。
conda install tensorflow-gpu

顯然如下圖,依賴是cuda10的,不知道能不能生效。

開始安裝后,由于網(wǎng)絡(luò)帶寬會中斷,多執(zhí)行幾次,conda配置了源,不會太久。

- 使用之前的驗證方式還是出問題了!!
查看原因,不支持高版本,除非更新驅(qū)動,就吃裝cuda


- 突然想起我裝的是ubuntu18.04雙系統(tǒng),后來就沒有搞win了,哈哈哈,然后在ubuntu上去學(xué)習(xí)了



