PyTorch(一):简介、安装

1、简介

PyTorch是一个基于Python的库,用来提供一个具有灵活性的深度学习开发平台。PyTorch的工作流程非常接近Python的科学计算库——numpy。

2017年1月18日,facebook下的torch7团队宣布Pytorch开源,官网地址:pytorch。

2018年4月25日,PyTorch 官方发布 0.4.0 版本,该版本的PyTorch 有多项重大更新,其中最重要的改进是支持 Windows 系统。

Windows用户能直接通过conda、pip和源码编译三种方式来安装Pytorch,不过需要强调Windows下的Pytorch仅支持Python3.5和Python3.6,不支持其他的Python3版本,也不支持Python2。

PyTorch可以顺利地与Python数据科学栈集成。

PyTorch为我们提供了一个框架,以便可以在运行时构建计算图,甚至在运行时更改它们。在不知道创建神经网络需要多少内存的情况下这非常有价值。

PyTorch的其他一些优点还包括:多gpu支持,自定义数据加载器和简化的预处理器。

开始使用PyTorch时应该了解的主要元素:PyTorch张量、数学运算、Autograd模块、Optim模块、神经网络模块

2、安装

pytorch官网
根据自动计算机情况,自动配置下载命令
在这里插入图片描述
我是AMD显卡,因此CUDA选择了NONE

pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio===0.7.2 -f http

执行过程和结果如下图所示:
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_47542175/article/details/113857530