簡介
PyTorch
是一個(gè)Python
深度學(xué)習(xí)框架淑仆,旨在實(shí)現(xiàn)簡單靈活的實(shí)驗(yàn)候衍,其提供兩個(gè)高級功能:
- 具有強(qiáng)大的 GPU 加速的張量計(jì)算(如
NumPy
) - 包含自動求導(dǎo)系統(tǒng)的的深度神經(jīng)網(wǎng)絡(luò)
考慮到Python
在計(jì)算科學(xué)領(lǐng)域的領(lǐng)先地位,以及其生態(tài)完整性和接口易用性还棱,幾乎任何框架都不可避免地要提供Python
接口。在2017年慨绳,Torch
的幕后團(tuán)隊(duì)推出了PyTorch
爱榔。PyTorch
不是簡單地封裝Lua Torch
提供Python
接口被环,而是對Tensor
之上的所有模塊進(jìn)行了重構(gòu),并新增了最先進(jìn)的自動求導(dǎo)系統(tǒng)详幽。
自 2017 年初首次推出筛欢,PyTorch
很快成為 AI 研究人員的熱門選擇并受到推崇。PyTorch
有許多優(yōu)勢妒潭,如采用Python
語言悴能、動態(tài)圖機(jī)制揣钦、網(wǎng)絡(luò)構(gòu)建靈活以及擁有強(qiáng)大的社群等雳灾。由于其靈活、動態(tài)的編程環(huán)境和用戶友好的界面冯凹,PyTorch
是快速實(shí)驗(yàn)的理想選擇谎亩。
目前,PyTorch
也廣泛的應(yīng)用在學(xué)術(shù)界與工業(yè)界宇姚。
安裝
PyTorch
的安裝可以參考官方的QUICK START 匈庭,其針對操作系統(tǒng)和軟件環(huán)境提供了多種不同的安裝方式。并且網(wǎng)頁會自動檢測你的軟件環(huán)境浑劳,生成最適合的安裝命令阱持。
根據(jù)提示,我只需要在終端當(dāng)中輸入對應(yīng)的指令就可以了魔熏,當(dāng)然我們也可以手動點(diǎn)擊選項(xiàng)來生成不同的命令衷咽。由于conda
源下載過于緩慢,我們換用pip
的方式進(jìn)行安裝蒜绽。
其中pytorch
是主模塊镶骗,torchvision
是輔助模塊。
conda
安裝:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
pip
安裝:
pip3 install https://download.pytorch.org/whl/cu90/torch-1.0.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision
安裝結(jié)果如下所示:
測試PyTorch
是否能夠正常運(yùn)行: