Why choose pytorch
在深度學(xué)習(xí)領(lǐng)域蘸秘，一般用于開(kāi)發(fā)的主流深度學(xué)習(xí)框架包括：Caffe, TensorFlow, MXNet, pytorch。除此之外國(guó)內(nèi)還有百度的paddpaddle等。筆者只用過(guò)Caffe,淺淺了解TensorFlow。 Caffe是一個(gè)優(yōu)秀開(kāi)源深度學(xué)習(xí)框架，但是目前早已經(jīng)停止更新，而且caffe使用了許多第三方庫(kù)霜第，每次安裝編譯caffe都不是一帆風(fēng)順。TensorFlow只是簡(jiǎn)單了解户辞，有許多自創(chuàng)的概念比如Session, Variable等泌类， 寫(xiě)起來(lái)很啰嗦。
pytorch是Facebook的開(kāi)源深度學(xué)習(xí)框架底燎，背后有大公司的支持刃榨，而且有活躍的pytorch社區(qū)弹砚，文檔也很齊全，對(duì)于開(kāi)發(fā)者而言很友好枢希。
開(kāi)發(fā)環(huán)境
- Ubuntu 18.04
- Anaconda3
- pycharm
pytorch 安裝
首先進(jìn)入pytorch官網(wǎng)：
image.png
根據(jù)自己電腦的環(huán)境桌吃，選擇不同的版本：
image.png
因?yàn)楸救碎_(kāi)發(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
說(shuō)明pytorch, torvision安裝成功苞轿。
