概述

PyTorch API 文档

PyTorch 是什么

PyTorch 是一个提供两个高级功能的 python 包:

1. 具有强 GPU 加速度的张量计算 (如 numpy)
2. 深层神经网络建立在基于磁带的自动调整系统上

注: PyTorch 中的 Torch 和 TensorFlow 中的 Tensor 是一个意思.

Why? PyTorch

Pytorch 在近几年发展迅速, 如图:

PyTorch 作为一个主流的深度学习框架已经一步步和 TensorFlow 齐名. 比起 TensorFlow 1, PyTorch 的使用会来的更简单.

对深度学习没有了解的同学可以看一下这个:

TensorFlow2 专栏

安装

Windows + R 打开命令执行符:

CPU 版本安装:

pip install torch==1.8.0+cpu torchvision==0.4.1+cpu -f https://download.pytorch.org/whl/torch_stable.html

GPU 版本安装:

pip install torch===1.8.0 torchvision===0.4.1 -f https://download.pytorch.org/whl/torch_stable （默认是CUDA10版本）

参看是否安装成功:

import torch# 输出版本
print(torch.__version__)

输出结果:

1.9.0+cpu