1.优点
可以自动计算梯度
调试方便
动态图
2.安装
我安装的是pip管理,cuda8.0,python2.7
安装的代码为
pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp27-none-linux_x86_64.whl
pip install torchvision
这里是库安装,应该不能修改层的代码,但是可以增加层