TensorFlow的版本選擇和安裝


一、說在前面

TensorFlow是一個基於數據流編程(dataflow programming)的符號數學系統,被廣泛應用於各類機器學習(machine learning)算法的編程實現,其前身是谷歌的神經網絡算法庫DistBelief。
Tensorflow擁有多層級結構,可部署於各類服務器、PC終端和網頁並支持GPU和TPU高性能數值計算,被廣泛應用於谷歌內部的產品開發和各領域的科學研究。

二、選擇版本

(1)選擇CPU還是GPU

TensorFlow在安裝時首先面臨的就是版本選擇的問題:CPU版還是GPU版?版本號選哪個

判斷條件 結果
顯卡是否NVIDIA系列? 是=GPU;否=CPU
若是NVIDIA系列,計算能力如何? 大於等於3.5=GPU;小於3.5=CPU


首先,查看自己電腦顯卡的型號。如果顯卡是NVIDIA系列的,繼續下面步驟;如果顯卡不是NVIDIA系列的,直接裝CPU版。

  右鍵我的電腦/屬性/設備管理器/顯示適配器,這是我的

  

然后,如果是NVIDIA系列的,則查詢該顯卡的計算能力。詳見:查詢顯卡計算能力。點開自己顯卡對應的系列,查看自己顯卡的GPU計算能力(即,Compute Capability)。如下圖,顯卡NVIDIA GeForce RTX 1050對應的計算能力為6.1,可以使用GPU

 

最后,到官網查詢發布的GPU支持,查看硬件要求。本文查詢時的標准是3.5(如下圖)。如果計算能力≥3.5,可以裝GPU版;相反<3.5的只能選擇CPU版了。

(2)選擇版本號

選擇完GPU還是CPU就可以選擇相應的版本號了,兩個的區別就是運算速度的問題,通過看別人的博客感覺GPU的比較麻煩還是選擇了CPU,怪我太懶T_T,如果有伙伴還是想要有更好的體驗可以參考這篇博客

三、安裝TensorFlow(我裝的是CPU版本的)

(1)進入Anaconda的Promot

 

 

(2)為TensorFlow創建虛擬環境

 

在promot里面輸入 conda create -n tensorflow python=3.5

 

可以通過 conda info -e來查看自己創建的虛擬空間

 

 

(3)進入虛擬空間tensorflow

在promot里面輸入 activate tensorflow,如果切換了就說明創建成功了、

我們要安裝的是CPU版本,那么在命令下緊接着輸入:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow

 如果出現黃色字體提示pip的版本不夠新按照要求輸入命令即可

 

四、測試安裝結果

在tensorflow虛擬空間里面輸入python后,依次輸入

import tensorflow.compat.v1 as tf

tf.compat.v1.disable_eager_execution()

sess = tf.Session()

= tf.constant(10)

b= tf.constant(12)

sess.run(a+b)

如果輸出結果是22則說明安裝成功了!

 

 五、安裝失敗還可以參考

安裝:https://www.cnblogs.com/ming-4/p/11516728.html

 

 

 

 

 

 

 

 

 

 

 

 

 


免責聲明!

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



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