torch 作為和tensorflow齊名的深度學(xué)習(xí)框架,并且與tensorflow不一樣的動(dòng)態(tài)網(wǎng)絡(luò)圖分预,tensorflow是靜態(tài)的减途,pytorch作為python版本的torch真是越用越順手种玛,而且有facebook團(tuán)隊(duì)專(zhuān)門(mén)維護(hù),開(kāi)源至今1年多的時(shí)間圈粉無(wú)數(shù)致燥。我也計(jì)劃主要的深度學(xué)習(xí)登疗,強(qiáng)化學(xué)習(xí)的大部分工作漸漸由tensorflow遷移到pytorch。
如果是在ubuntu或者osx上安裝嫌蚤,參見(jiàn)官網(wǎng)辐益,自己選擇os和python版本以及是否有cuda,選擇好以后copy其代碼就行:pytorch
pytorch目前官方?jīng)]有windows版本放出脱吱,如果想在windows上玩pytorch怎么辦智政?我們有conda呀。箱蝠。续捂。。
# for CPU only packages
conda install -c peterjc123 pytorch-cpu
# for Windows 10 and Windows Server 2016, CUDA 8
conda install -c peterjc123 pytorch
# for Windows 10 and Windows Server 2016, CUDA 9
conda install -c peterjc123 pytorch cuda90
# for Windows 7/8/8.1 and Windows Server 2008/2012, CUDA 8
conda install -c peterjc123 pytorch_legacy
另外torchvision的安裝:
pip install torchvision