書名: 深度學習與圖像識別:原理與實踐作者名: 魏溪含 涂銘 張修鵬本章字數: 394字更新時間: 2019-08-15 18:06:35
2.1.5 PyTorch
PyTorch是一個Python優先的深度學習框架,能夠在強大的GPU加速的基礎上實現張量和動態神經網絡。
PyTorch是一個Python軟件包,其提供了兩種高層面的功能,具體如下。
1)使用強大的GPU加速的Tensor計算(類似于Numpy)。
2)構建基于tape的autograd系統的深度神經網絡。
3)活躍的社區:PyTorch提供了完整的文檔,循序漸進的指南,作者親自維護論壇以供用戶交流和求教問題。Facebook人工智能研究院(FAIR)對PyTorch提供了強力支持,作為當今排名前三的深度學習研究機構,FAIR的支持足以確保PyTorch獲得持續的開發更新,而不至于像許多由個人開發的框架那樣曇花一現。
如有需要,你也可以復用你喜歡的Python軟件包(如Numpy、scipy和Cython)來擴展PyTorch。
相對于Tensorflow,PyTorch的一大優點是,它的圖是動態的,而Tensorflow等都是靜態圖,不利于擴展。同時,PyTorch非常簡潔,方便使用。本書將選取PyTorch作為圖像識別的主要實現框架。
總結:如果說TensorFlow的設計是“Make It Complicated”,Keras的設計是“Make It Complicated And Hide It”,那么PyTorch的設計真正做到了“Keep it Simple,Stupid”。
項目地址:http://pytorch.org/。