Anaconda安裝Tensorflow-gpu(2020.7)


寫在前面:由於電腦之前被自己鼓搗了一下不小心把開機引導刪了,重裝了系統,所以好多環境都得重新配置起來。今天來弄一下conda的環境。

1.安裝Anaconda

Anaconda的安裝沒什么好說的,正常到官網下載安裝包。

地址:https://www.anaconda.com/products/individual

找到自己需要的版本點擊下載。

卸載電腦上已有的python

先看下自己電腦上安裝的python版本

我這里是3.8.3,為了以防后面的一些奇怪的版本不兼容問題,我先把這個卸載。(如果你是3.7的話就不用這一步了)

直接在官網找到相應版本的安裝包(記住是安裝包,不是直接解壓就用的那種),然后運行uninstall.

卸載完成,然后我們開始裝conda.

開搞conda

一路next,這邊選擇安裝磁盤地址自己決定,我是不改了。

然后這一步:

這里我是直接勾選添加到環境變量中去,不用我自己再配置路徑了。可是如果你電腦里有多個版本的話,不建議勾選,容易出現問題。不過想想看這種安裝教程的肯定是小白,怎么還會考慮那么多東西,一步步跟着來就完事了。

安裝!

就等着吧,過會就好了。

我們打開cmd看一下python版本:

python -V   //這里如果單輸入python的話是進入python環境,可以看到這個python是在conda環境下的

2.安裝Tensorflow-gpu

安裝這個GPU版本真的是一件很費力的事情,你得把對應的版本全部校對好,否則版本不匹配照樣還是用CPU處理。

我們打開conda的prompt

現在我們所處的環境就是在conda下,裝的各種環境也是在conda下。千萬不要用cmd!!!

出於對conda的禮貌,我們先問候一下它的版本。

conda -V 

嗯,挺好的,那就來安裝吧。這邊安裝的話由於是國外的服務器,所以下載過程中容易中斷無響應。所以這里要更改一下國內的鏡像源,當然了如果不改鏡像的話也可以,我是沒改(doge):

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

然后在新建一個虛擬環境:

conda create -n TC2 python=3.7  //新建一個python3.7的環境,命名為TC2

等待安裝完畢

進入該環境:

conda activate TC2

有這個顯示就說明我們進入了這個TC2的環境,之后我們就在這個環境下安裝tensorflow-gpu

conda install tensorflow-gpu //這里也可以指定版本,比如conda install tensorflow-gpu==2.1.0

輸入y確認。我們不需要再額外安裝cuda和cudnn,因為他已經包含在安裝的列表里了。

我這里安裝的是2.1.0的版本,靜靜等待安裝完成,如果安裝出現問題,比如進度卡住了之類的多半是網絡不穩定,可以選擇切換到國內鏡像。

等待完成。然后我們可以看一下有沒有安裝完成:

conda list  //查看當前環境下安裝的包

已經成功安裝了tensorflow-gpu2.1.0的版本。

為了操作的方便,我們需要在該環境下安裝spyder和jupyter,這個看個人喜好,反正我兩者都用,jupyter的話可以一步一步寫,運行就得到結果,spyder的話可以查看變量,非常方便。哪個方便用哪個,不多說了。然后我們打開jupyder,由於jupyder直接打開的話默認目錄是在安裝目錄下的,所以我們需要用簡單的命令定位到我們要新建文件的目錄下,比如這邊我在桌面新建了一個TC2的文件夾:

cd desktop
cd TC2

這樣就到我們所選擇的目錄下了,然后:

jupyter notebook //運行jupyter 

我們新建一個python3的腳本,輸入:

import tensorflow as tf
tf.__version__  

查看當前tensorflow版本

import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

查看可用的GPU個數

如果你確定自己的GPU是正常的,但是運行結果是0,那就得懷疑一下自己的版本問題,一般來說版本只要符合要求,即Python版本,tensorflow-gpu版本,cuda和cudnn匹配。具體參照tensorflow官網

3.總結

以上就是本次安裝Anaconda和tensorflow-gpu的教程,一般來說跟着做就沒什么問題了。而且現在的安裝越來越傻瓜式了,比如添加路徑什么的都能自動添加,依賴包能捆綁安裝了。方便了不少,所以這次我基本沒踩什么坑。有什么問題大家可以提出來一起解決。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM