PyTorch是深度学习的主流框架之一,新手入门相对容易。PyTorch是一个开源的Python机器学习库,其前身是2002年诞生于纽约大学 的Torch。它是美国Facebook公司使用python语言开发的一个深度学习的框架,2017年1月,Facebook人工智能研究院(FAIR)在GitHub上开源了PyTorch。
{
"cells": [
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# PyTorch登场"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Torch是什么?一个火炬!其实跟Tensorflow中Tensor是一个意思,当做是能在GPU中计算的矩阵就可以啦!如果没玩过Tensorflow,Numpy总用过吧,也可以当做是ndarray的GPU版!"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"![title](./img/1.png)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## PyTorch的飞速发展\n",
"- 框架好不好用,咱们来看图说话!"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"![title](./img/3.png)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"![title](./img/4.png)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"PyTorch可以说是现阶段主流的深度学习框架了,武林盟主之争大概是这个历史。。。15年底之前Caffe是老大哥,随着Tensorflow的诞生,霸占江湖数载,19年起无论从学术界还是工程界PyTorch已经霸占了半壁江山!"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"### 安装PyTorch:使用PIP的方法比较简单"
]
},
{
"cell_type": "markdown",
"metadata": {
"collapsed": true
},
"source": [
"CPU版本安装:pip install torch==1.3.0+cpu torchvision==0.4.1+cpu -f https://download.pytorch.org/whl/torch_stable.html\n",
"\n",
"GPU版本安装:pip install torch===1.3.0 torchvision===0.4.1 -f https://download.pytorch.org/whl/torch_stable (默认是CUDA10版本)"
]
},
{
"cell_type": "code",
"execution_count": 1,
"metadata": {},
"outputs": [
{
"data