1. 程式人生 > 程式設計 >windows系統Tensorflow2.x簡單安裝記錄(圖文)

windows系統Tensorflow2.x簡單安裝記錄(圖文)

自從接觸了python,再到機器學習和深度學習,要學習的東西向越拉越多了!!!因為課題的需要接觸了tensorflow,我直接就是一個好傢伙,光安裝就看了好多教程,裝了好幾天,回頭看看其實就幾句命令就可以了,對小白來說太不友好了。用習慣了windows系統,換成其他系統的話有需要學好久。

這裡以windows10+anaconda+tensorflow2.3+python3.8為例。

第一步:下載環境管理軟體

Anaconda和Miniconda兩個自選(其實就是完整版和精簡版的區別!),官網下載即可,成年人當然選擇最新版本了。
anaconda
連結: link.
miniconda
連結: link.

下載好之後開啟,進入正題咯!

在這裡插入圖片描述

點選左下角“create”建立一個虛擬環境,名稱自取(但是一定要記住,牢記!),這裡以“tensorflowceshi”為例,選擇python3.8(3.5-3.8版本任選)。

建立好如下圖:

在這裡插入圖片描述

然後開啟軟體下的命令列Anaconda Prompt,可以看到現在的環境是預設的base,這裡輸入下面語句(這一步是為了啟用剛才建立的虛擬環境,以後需要這個環境的時候都需要activate一下):

activate tensorflowceshi

效果如圖:

在這裡插入圖片描述

第二步:安裝Tensorflow

安裝的時候可能會提示pip版本不對,根據它的提示升級一下就可以啦。(因為pip版本要求大於19.0才可以安裝)

1.CPU版本

接著上面的步驟,activate之後在命令列中輸入:(這裡選擇安裝tensorflow2.3版本,從豆瓣源安裝,速度超級快,誰用誰知道):

pip install tensorflow-cpu==2.3.0 -i https://pypi.douban.com/simple/

2.GPU版本

GPU版本的安裝才是最煩人的,空守著電腦配置卻用不了,哈哈。

劃重點:NVIDIA顯示卡、顯示卡驅動>418.X、CUDA版本為10.1、cudnn版本>7.6,可以通過nvidia-smi進行檢視,近兩年剛買的電腦的話一般是都滿足要求的。(也可以直接進行下面的安裝,不成功再回來找原因)

接安裝cpu版本步驟,或者重新進命令列activate後,在命令列輸入:

conda install cudatoolkit=10.1

在這裡插入圖片描述

執行到這裡輸入y,回車:

在這裡插入圖片描述

上一步完成後,繼續在命令列輸入:

conda install cudnn=7.6.5

在這裡插入圖片描述

同樣在過程中需要輸入y,回車進行確認。

接著最後一步,繼續輸入:

pip install tensorflow-gpu==2.3.0 -i https://pypi.douban.com/simple/

到這裡就安裝成功了,是不是很開心?

在這裡插入圖片描述

後面需要什麼包(package)就用pip安裝就可以了,像matplotlib、notebook等等常用的,推薦還用上面的豆瓣源,速度超級快,其他跟環境相關的都是conda xx,不要搞混了。

安裝好之後怎麼用呢?

可以在activate環境之後開啟jupyter notebook;
也可以在pycharm中將環境設定為剛剛建立的環境就可以了。

總結:

巴拉巴拉說了這麼多,其實只需要幾行程式碼就可以安裝成功了,操作一次就可以受用一輩子!!!寫給 自己,也寫給在深度學習路上掙扎的小白!

到此這篇關於windows系統Tensorflow2.x簡單安裝記錄(圖文)的文章就介紹到這了,更多相關Tensorflow2.x 安裝內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!