TensorFlow的静态图方式不易于灵活部署网络,于是开始转战PyTorch。
PyTorch 具有和 NumPy 直接交互的良好特性,适合敏捷部署神经网络,提高实验工作效率。
本文开始记录 PyTorch 学习心得与一些关键问题。
1. 开始与安装
PyTorch 官网.
PyTorch GitHub 主页.
配置 Nvidia GPU 环境请参考这里。
PyTorch 官网提供了非常仔细的安装指导,我喜欢在 Python 虚拟环境中工作。
Linux
(virtualenv)$ pip install torch torchvision
Windows
(virtualenv)$ pip install https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-win_amd64.whl
(virtualenv)$ pip install torchvision
2. 官方教程
PyTorch 官方教程.
2.1 Getting Started
Deep Learning with PyTorch: A 60 Minute Blitz