- 從零開始學TensorFlow2.0
- 趙銘 歐鐵軍編著
- 337字
- 2020-09-25 12:47:39
1.3 常用的深度學習框架
選擇一個合適的深度學習框架非常重要,每個深度學習框架都有其自身的特點,根據這些特點進行學習,能夠起到事半功倍的效果。
最流行的深度學習框架有TensorFlow和PyTorch,具體情況如下。
(1)TensorFlow是使用人數最多、社區最龐大的框架,由Google公司開源,維護與更新比較頻繁,并且有Python和C++的接口,教程也非常完善,是深度學習的主流框架之一。2019年3月,TensorFlow發布了TensorFlow 2.0,與TensorFlow 1.x版本相比,其做了很多改變,比較重要的有:①重用Keras可以使模型的構建和運行更加簡單;②強調各平臺的通用性;③與TensorFlow 1.x相比,TensorFlow 2.0提供了更方便的模型保存和調用方式。
(2)PyTorch是從Torch框架演變來的深度學習框架,它使用Python在Torch框架上寫了很多內容。不僅更加靈活,支持動態圖,還提供了Python接口。PyTorch由Torch7團隊開發,能夠實現強大的GPU加速,且支持動態神經網絡。