【入门】PyTorch

目录

• 1. 安装
• 2. 入门
  • 2.1. 入门示例程序

1. 安装

homepage: get-started

默认安装的是 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. 入门

homepage

• 数据结构：张量
• CUDA张量
• 转换为NumPy数组

2.1. 入门示例程序

homepage