win10下anaconda3 搭建 tensorflow-gpu(總結(jié))

注:此博客為不需要安裝cuda和cudnn。(直接搭建tensorflow-gpu環(huán)境)
注:當(dāng)前顯卡GTX1050。(當(dāng)前文章寫作測試日期2020年4月22日)

一、anaconda安裝

  1. 官方下載
  2. 配置conda鏡像源(主要一下主要安裝用它)

https://mirror.tuna.tsinghua.edu.cn/help/anaconda/

  1. 配置pip鏡像源

臨時使用: https://mirror.tuna.tsinghua.edu.cn/help/pypi/
配置(注意看評論區(qū)): https://developer.aliyun.com/mirror/pypi?spm=a2c6h.13651102.0.0.3e221b114TOXjb

二、配置查看

  1. 打開NVIDIA控制面板查看當(dāng)前電腦支持的CUDA。


    image.png
  2. 查看源中tensorflow-gpu版本


    image.png
  3. 查看gpu對應(yīng)的其他依賴包版本

查看官網(wǎng)鏈接:https://www.tensorflow.org/install/source#common_installation_problems

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

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


安裝一個py36的環(huán)境

執(zhí)行安裝:


image.png

查看得知安裝依賴所需的cudatoolkit=9.0.1,低于我電腦支持的9.1非常不錯。那就選擇y繼續(xù)。如果一次沒有安裝上,多執(zhí)行自己安裝命令。conda可是換了源的。
后面還有測試GPU是否生效,別著急。
也順便總結(jié)可能出現(xiàn)的如第6錯誤:
  1. 常見錯誤:

    • 創(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
  • 安裝沒有指定版本的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

image.png

二、安裝

測試安裝是否成功

  1. 出現(xiàn)報錯:


    image.png
  2. 解決報錯:

參考1-源碼解決(注意查看評論):https://blog.csdn.net/bigdream123/article/details/99467316
參考2-降低numpy版本(使用conda):https://blog.csdn.net/kobe_academy/article/details/99706595

  1. 開始解決:
image.png
  1. 再次測試是否gpu生效:

參考: https://blog.csdn.net/weixin_43522055/article/details/99617967

image.png

至此大功告成。開始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?先試試吧。

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

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


image.png

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


image.png
  1. 使用之前的驗證方式還是出問題了!!

查看原因,不支持高版本,除非更新驅(qū)動,就吃裝cuda


image.png
image.png
  1. 突然想起我裝的是ubuntu18.04雙系統(tǒng),后來就沒有搞win了,哈哈哈,然后在ubuntu上去學(xué)習(xí)了
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容