最近有好多小夥伴想玩玩pytorch。那索性寫個教程。算是個pytorch的入門。後面如何玩的溜還是得看個人。
STEP 0 PYTORCH是什麼
首先。。。pytorch是一個深度學習的架構，可以像一個python庫一樣使用。避免了你手撸各種NN推理和訓練代碼的尴尬。。。你隻需要直接用pytorch，上面寫好了很多常用的東西。
STEP 1 如何安裝Pytorch
這個東西可比當年安裝早版本的caffe要友善太多了。。。當年caffe裝的要吐血。這個安裝就兩步。簡單明了。
step1.1 安裝一個conda pythonAnaconda Python/R Distribution - Free Downloadwww.anaconda.com
這個是一種程式設計語言。但是按時需要你來安裝到計算機上。python這個語言很神奇。。。有很多版本是以記得别搞錯版本。為了友善版本管理。這裡了直接安裝anaconda來管理python...
上面那個網站。。點進去往下翻，直接點python 3.7下面的下載下傳。（Download）。這個大約四百多兆。
一路點下一步。中間要是想改安裝位置可以改。也可以不改無腦下一步。
裝完以後還需要加上環境變量。。。賊麻煩。
如何添加環境變量：右鍵我的電腦。屬性。進階系統設定。環境變量。在系統變量裡找到Path. 輕按兩下。
系統變量在下面。。。
添加兩個環境變量。。。剛才你把anaconda裝哪兒了這個地方就添加什麼哪兒。
打開powershell.. 輸入python -V. 出現上面這種狀态就說明你裝成功了。。。
step1.2 安裝一個pytorchPyTorchpytorch.org
打開這個網站。。。點Get Started. 到了這個界面。選擇你現在的配置。
需要注意的是language要選對。Python 3.7如果你按照我寫的安裝的話。。
CUDA那個是的顯示卡配置。這個需要看你裝了哪個版本的CUDA。機器上沒有顯示卡。可以選沒有裝。
如果有顯示卡，點看控制台VNIDIA控制台幫助系統資訊元件NVCUDA.DLL .如下圖所示。
然後你看到那一行Run this Command了麼。。
複制它。然後以管理者身份打開一個powershell.
然後咣的一聲。。。出error了。。你說你不出error. 那也沒問題。跳過下面一步。
出現這辣雞問題的原因可能是網絡聯不通。。。你需要添加一個清華的源來下載下傳（有段時間清華的源conda GG了，不知道現在行不行）。輸入下面4條。。可以切換源。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
正确的安裝界面是這樣的。。。等着吧。親測要是不換源非常慢。。。50幾兆的東西，建議配置一下科學上網。
安裝完以後，輸入
python
import torch
import torchvision
出現上面這些東西，沒報錯，意味着安裝成功了。。。
STEP 3 來個Mnist練練手
下面搭建一個祖傳的mnist。這個屬于神經網絡的hello world.
建立一個檔案，字尾名改成py.
檔案裡先導入一堆torch的玩意兒。
然後把mnist的資料集載入進來。這個地方mnist資料集不需要提前下載下傳，如果檢測到路徑下沒有會自動下載下傳。
接下來把網絡定義後。我們用個3層CNN就可以搞定mnist.
接下來要寫上訓練函數。。。求導不需要自己寫。但是訓練步驟還是得自己寫的。如下。
然後接下來寫前向測試函數
最後寫上執行函數。。。為了示範。隻訓練了1個epoch. 實際上要訓練若幹。
這樣就完成了。。如果你懶得敲代碼。可以直接複制下面的代碼。
import
最後一步。運作你的代碼。
打開powershell. 輸入python 你的檔案名.py
會看到下面的場景
嗯。。到這一步你就可以跑起來一個基本的網絡了。。。
其他網絡也可以用類似的方法搞出來。
後記
到這個地方這篇文章就完成了。
無論是pytorch、caffe、tensorflow都隻是一個工具。主要的門檻還是如何用他們搭起來可以實際使用的網絡模型。是以不比糾結用哪個。
本次主要介紹的是pytorch. 其使用方法還是很簡單的。不過現在幾個平台吧。caffe比較老了，tensorflow 還有1.0和2.0. 比較亂。pytorch吧，總更新，和python一樣比較煩。不過感覺上還是pytorch兼顧靈活性的同時用法比較一以貫之。算是學習深度學習比較好的平台。
就是這樣。