1. 安装
默认安装的是 GPU-CUDA10.2 版本
pip install torch torchvision
如果是CPU：
pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
Windows/Mac平台或其他类型的CUDA，请参考官网。
2. 入门
- 数据结构：张量
- CUDA张量
- 转换为NumPy数组
默认安装的是 GPU-CUDA10.2 版本
pip install torch torchvision
如果是CPU：
pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
Windows/Mac平台或其他类型的CUDA，请参考官网。