Article / 文章中心

Win10+Python3.6下Pytorch安装(基于conda或pip)

发布时间:2021-12-08 点击数:504

Pytorch特点及优势

2.1 Pytorch特点

PyTorch提供了运行在GPU / CPU之上,基础的张量操作库;

可以内置的神经网络库;

提供模型训练功能;

支持共享内存的多进程并发(multiprocessing)库等;

2.2 Pytorch特点

处于机器学习第一大语言Python的生态圈之中,使得开发者能使用广大的Python库和软件;如NumPy,SciPy和Cython(为了速度把Python编译成C语言);

(最大优势)改进现有的神经网络,提供了更快速的方法 - 不需要从头重新构建整个网络,这是由于PyTorch采用了动态计算图(动态计算图)结构,而不是大多数开源框架( TensorFlow,Caffe,CNTK,Theano等)采用的静态计算图;

提供工具包,如火炬,torch.nn,torch.optim等;

安装Pytorch

Pytorch官网https://pytorch.org/

image.png

安装命令如下安装命令如下

conda install pytorch-cpu -c pytorch

image.png

测试Pytorch

#启动Ipython ipython #导入Pytorch包 import torch#启动Ipython ipython #导入Pytorch包 import torch