TensorFlow 2 教學:安裝 TensorFlow 2

tensorflow

前言

TensorFlow2 最近出了 RC 版本 (Release Candidate) (2020年7月更新,目前版本已經來到tensorflow2.2, tensorflow2.3在RC階段 ) 今天來教大家如何安裝 TF2 CPU版本,GPU版本比較複雜,需安裝 Cuda,之後再發文章給大家教學。若想體驗一下 Deep Learning、TensorFlow,CPU版本是很足夠的,簡單上手。

第一步 建立 Conda Environment

我們可以先創一個 Conda Environment 來管理套件,請見 : Python教學:Anaconda Navigator 或者是使用以下的指令:

使用conda創建一個環境,名稱叫"tf2":
$ conda create --name tf2

activate 剛剛創的環境:
$ conda activate tf2

(2023/10/20 更新,原始文章使用 source activate tf2 而不是 conda activate tf2,source activate 是舊的使用方式,理論上也是可以成功啟動虛擬環境,但 conda 建議使用 conda activate 因此更新文章,謝謝網友在底下留言提醒)

第二步 安裝 TensorFlow2

依照 TensorFlow 官網的指示,使用 pip 安裝 TensorFlow:

$ pip install tensorflow  

如果噴 Error,可能代表 pip 版本太舊,先給他更新一下~

檢查一下 pip 版本,pip18 太舊了,需要升級 pip:
$ pip --version
-> pip 18.1

升級pip:
$ pip install --upgrade pip

安裝 TensorFlow2:
$ pip install tensorflow  

第三步 測試 Tensorflow

直接進入 Python 測試
$ python

>>> import tensorflow as tf # 如果沒有噴Error就代表安裝成功!
>>> print(tf.add(1, 2)) 

# 會得到 tf.Tensor(3, shape=(), dtype=int32)
# 看到 3 這個數字代表完成了最基本的加法!

2 Comments

    1. 似乎在 conda 4.4 版本後,conda 期待使用者使用 conda activate 而不是 source activate 來使用虛擬環境。謝謝提醒,我來更正文章。

留言討論區