Tensorflow簡單實踐系列(一):安裝和運行


TensorFlow 是谷歌開發的機器學習框架。

安裝 TensorFlow

直接使用 pip 安裝即可,添加豆瓣鏡像可以加快速度:

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

如果有 GPU 可以充分利用,安裝:

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

目前我使用的 TensorFlow 版本是  tensorflow==1.14.0 ,目前 TensorFlow 不支持 3.7 及以上的版本。我的 Python 版本使用的是 3.6

認識 TensorFlow 包

TensorFlow 包有兩個主要的目錄:corecontrib。core 是包含核心模塊的目錄,contrib 是尚未歸入到 core 的目錄(可能隨時會歸入)。

認識其中重要的包名:

  • tensorflow:TensorFlow 的主包
  • tf.train:優化器、與訓練有關的類
  • tf.nn:神經網絡類及其數學運算
  • tf.layer:多層神經網絡的相關函數
  • tf.contrib:包含不穩定或者實驗性的代碼
  • tf.image:圖像處理函數
  • tf.estimator:提供訓練和評估的高級抽象的工具
  • tf.logging:記錄日志的工具
  • tf.summary:生成總結的工具
  • tf.metrics:評估機器學習結果的函數

跑一個 TensorFlow 應用

說了這么多,寫點實際的代碼跑一下 TensorFlow 應用。

1 import tensorflow as tf
2 
3 # 創建一個張量
4 msg = tf.string_join(["Hello ", "TensorFlow!"])
5 # 發起一個會話
6 with tf.Session() as sess:
7     print(sess.run(msg))

輸出:

b'Hello TensorFlow!'

TensorFlow 中張量用於存儲數據,而會話用於真正執行數據運算操作。

 


免責聲明!

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



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