torch 作为和tensorflow齐名的深度学习框架,并且与tensorflow不一样的动态网络图,tensorflow是静态的,pytorch作为python版本的torch真是越用越顺手,而且有facebook团队专门维护,开源至今1年多的时间圈粉无数。我也计划主要的深度学习,强化学习的大部分工作渐渐由tensorflow迁移到pytorch。
如果是在ubuntu或者osx上安装,参见官网,自己选择os和python版本以及是否有cuda,选择好以后copy其代码就行:pytorch
pytorch目前官方没有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