跳轉到內容

PyTorch/簡介

100% developed
來自華夏公益教科書

PyTorch,也稱為pytorch,是一個深度學習包。它也可以用於淺層學習,用於與深度學習無關的最佳化任務,以及用於使用或不使用 CUDA 的一般線性代數計算。

PyTorch 是眾多深度學習包之一。截至 2018 年 11 月,按貢獻者數量計算,它是 TensorFlow 之後排名第二的軟體包,按 GitHub 上的星數計算,它是 TensorFlow 和 Caffe 之後排名第三的軟體包 [1]。Keras 也應該在這裡提及。

PyTorch 來自於一種名為 Lua 的語言下的 Torch 包。因此,pytorch 在 python 中被稱為 torch。例如,import torchconda update pytorch

要安裝 PyTorch,請訪問其官方頁面 [pytorch.org]。不幸的是,您不能使用 sudo apt install 安裝 PyTorch。

優點和缺點

[編輯 | 編輯原始碼]

PyTorch 比 TensorFlow 更簡單、更易於學習,並且提供了更大的自由度。正如 Kirill Doubikov 所寫 [2]

總的來說,[PyTorch] 框架與 Python 語言更加緊密地整合,而且在大多數情況下感覺更原生。當您使用 TensorFlow 編寫程式碼時,您有時會感覺您的模型就像在一堵磚牆後面,只有幾個小孔可以進行通訊。

另一方面,他寫道

目前,TensorFlow 被許多研究人員和行業專業人士視為首選工具。該框架有很好的文件,如果文件不足,網際網路上有很多非常好的教程。您可以在 github 上找到數百個已實現和訓練的模型,從這裡開始。

與競爭對手相比,PyTorch 比較新(仍在測試版中),但它正在迅速獲得動力。文件和官方教程也很好。PyTorch 還包含幾種流行的計算機視覺架構的實現,這些實現非常易於使用。

截至 2019 年 9 月,PyTorch 已經不再是測試版了,但區別仍然存在。

TensorFlow 有一個很棒的視覺化工具,TensorBoard。從 1.1 版本開始,PyTorch 也完全支援 TensorBoard。


華夏公益教科書