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
执行过程和结果如下图所示: