PyTorch 與 TensorFlow 比較

最近學(xué)習(xí) PyTorch,參考網(wǎng)上大家的討論,簡單的將 PyTorch 和 TensorFlow 做了一下比較(2020/03/30),難免有疏漏,僅供參考

PyTorch 是后來者,但由于其易用性和靈活性,越來越收到青睞,隨著時(shí)間的推移,二者的比較應(yīng)該還會(huì)持續(xù)變動(dòng)

特性 PyTorch TensorFlow 備注
易用性 與 python 非常契合,與 numpy 非常類似,易于上手 定義圖的時(shí)候像是在使用一門新的語言,上手難
API 易于使用 混亂, 冗余
圖計(jì)算 動(dòng)態(tài)圖計(jì)算,可以在運(yùn)行時(shí)構(gòu)建計(jì)算圖,甚至可以在運(yùn)行時(shí)更改它們,非常靈活 Tensorflow 1.x 靜態(tài)圖構(gòu)造,因此需要對圖形進(jìn)行編譯,然后在執(zhí)行引擎上執(zhí)行 Tensorflow 2.0 默認(rèn)使用動(dòng)態(tài)圖,對 1.x 的兼容不太好,原有程序可能需要較大改動(dòng)
可視化 Visdom,TensorBoard TensorBoard
部署生產(chǎn) 開放式神經(jīng)網(wǎng)絡(luò)交換ONNX,解決通用深度學(xué)習(xí)框架之間的模型遷移部署 直接使用 TensorFlow serving 在 TensorFlow 中部署模型 一開始 pytorch 在部署上沒有優(yōu)勢,但隨著ONNX生態(tài)的發(fā)展,逐漸和 TensorFlow 拉平
調(diào)試 簡單,可以使用標(biāo)準(zhǔn)調(diào)試器 ,例如pdb或PyCharm TensorFlow eager 之前需要專門的調(diào)試器 tfdbg Tensorflow 2.0 默認(rèn)使用動(dòng)態(tài)圖以后和 PyTorch 一樣直接使用標(biāo)準(zhǔn)調(diào)試器
性能 不相伯仲
文檔 全面 全面

參考文章:

有些文章較早不能代表現(xiàn)在的情況

PyTorch攻勢兇猛,程序員正在拋棄TensorFlow?

PyTorch和TensorFlow哪家強(qiáng):九項(xiàng)對比讀懂各自長項(xiàng)短板

PyTorch VS TensorFlow誰最強(qiáng)?標(biāo)星15000+Transformers的運(yùn)行結(jié)果

Tensorflow eager execution 模式

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

  • PyTorch VS TensorFlow:細(xì)數(shù)兩者的不同之處 在今年 5 月初召開的 Facebook F8 開...
    Lornatang閱讀 20,403評(píng)論 11 6
  • 圍棋號(hào)稱人類最復(fù)雜的棋類運(yùn)動(dòng),但近兩年來,在AlphaGo的沖擊下,已經(jīng)潰不成軍。繼2016年AlphaGo以4:...
    金字塔下的小蝸牛閱讀 1,901評(píng)論 3 9
  • 今天,Caffe2和PyTorch突然成了熱點(diǎn)詞。。。深度計(jì)算的幾個(gè)引擎:Caffe2,PyTorch,Tenso...
    smilegator閱讀 1,509評(píng)論 0 1
  • 我今天背了古詩元日,宋,王安石,爆竹聲中一歲除,春風(fēng)送暖入屠蘇,千門萬戶曈曈日,總把新桃換舊符,主要是說一年過去了...
    高錦浩閱讀 93評(píng)論 0 0
  • 姓名 :秦飛 (單位)揚(yáng)州市方圓建筑工程有限公司 【日精進(jìn)打卡第46天】 哈爾濱363期 感謝一組 【知~學(xué)習(xí)】 ...
    fy秦飛閱讀 107評(píng)論 0 0

友情鏈接更多精彩內(nèi)容