一、Pytorch简介
Pytorch是2017年Facebook人工智能研究院团队在Github上开源的一个深度学习框架,其历史可追溯到2002年诞生于纽约大学的Torch。自发布之后,Pytorch的热度便持续高涨,随着深度学习持续火热,Pytorch这一深度学习框架被运用地越发广泛。
二、环境搭建
1、Anaconda安装
pytorch开发,推荐使用Anaconda,因为集成了很多相关的工具包,在Anaconda官网选择自己电脑的环境下载相关安装包即可安装。
2、 Pytorch安装
进入Pytorch的官网,根据自己的电脑环境配置选择,即可得到相关的安装的命令,之后在Anaconda Prompt中输入命令即可完成Pytorch的安装(cuda是有GPU的时候选择,使用CPU的话选择None即可)。
3 、进入命令窗口
执行
conda install pytorch torchvision cpuonly -c pytorch
等待安装完成,中途可能由于网络原因,报些HTTP错误,这个只要重新执行一下命令就行
三、验证torch安装
cmd进入窗口,
输入 python
再次输入
import torch
Torch=torch.rand(5,5)
print(Torch)
说明pytorch环境安装成功