Why choose pytorch
在深度學(xué)習(xí)領(lǐng)域，一般用于開發(fā)的主流深度學(xué)習(xí)框架包括：Caffe, TensorFlow, MXNet, pytorch。除此之外國(guó)內(nèi)還有百度的paddpaddle等。筆者只用過Caffe,淺淺了解TensorFlow。 Caffe是一個(gè)優(yōu)秀開源深度學(xué)習(xí)框架，但是目前早已經(jīng)停止更新，而且caffe使用了許多第三方庫(kù)，每次安裝編譯caffe都不是一帆風(fēng)順。TensorFlow只是簡(jiǎn)單了解，有許多自創(chuàng)的概念比如Session, Variable等， 寫起來很啰嗦。
pytorch是Facebook的開源深度學(xué)習(xí)框架，背后有大公司的支持，而且有活躍的pytorch社區(qū)，文檔也很齊全，對(duì)于開發(fā)者而言很友好。
開發(fā)環(huán)境
- Ubuntu 18.04
- Anaconda3
- pycharm
pytorch 安裝
首先進(jìn)入pytorch官網(wǎng)：
image.png
根據(jù)自己電腦的環(huán)境，選擇不同的版本：
image.png
因?yàn)楸救碎_發(fā)環(huán)境為Ubuntu, Anaconda3, python3.6因此在終端執(zhí)行： conda install pytorch torchvision -c pytorch
image.png
等待數(shù)分鐘就自動(dòng)安裝完畢。
在終端輸入：conda list
image.png
在終端進(jìn)入python,輸入：
import torch
import torchvision
image.png
說明pytorch, torvision安裝成功。
