Pytorch安装入门

一、什么是Pytorch
Pytorch的官网:https://pytorch.org/
PyTorch是一个较新的深度学习框架。从名字可以看出,其和Torch不同之处在于PyTorch使用了Python作为开发语言,所谓“Python first”。一方面,使用者可以将其作为加入了GPU支持的numpy,另一方面,PyTorch也是强大的深度学习框架。
二、Pytorch与TensorFlow对比

各个方面 Pytorch TensorFlow
上手 PyTorch本质上是Numpy的替代者,而且支持GPU、带有高级功能,可以用来搭建和训练深度神经网络。如果你熟悉Numpy、Python以及常见的深度学习概念(卷积层、循环层、SGD等),会非常容易上手PyTorch TensorFlow可以看成是一个嵌入Python的编程语言。你写的TensorFlow代码会被Python编译成一张图,然后由TensorFlow执行引擎运行。我见过好多新手,因为这个增加的间接层而困扰。也正是因为同样的原因,TensorFlow有一些额外的概念需要学习,例如会话、图、变量作用域(variable scoping)、占位符等
图创建和调试 PyTorch中简单的图结构更容易理解,更重要的是,还更容易调试。调试PyTorch代码就像调试Python代码一样。你可以使用pdb并在任何地方设置断点 调试TensorFlow代码可不容易。要么得从会话请求要检查的变量,要么学会使用TensorFlow的调试器(tfdbg)
全面性 PyTorch不支持的功能。它们是有沿维翻转张量(np.flip, np.flipud, np.fliplr)检查无穷与非数值张量(np.is_nan, np.is_inf)快速傅里叶变换(np.fft) 这些TensorFlow都支持。另外,TensorFlow的contrib软件包中,有更多PyTorch没有的高级功能和模型
序列化 PyTorch有一个特别简单的API,可以保存模型的所有权重或pickle整个类 TensorFlow序列化的主要优点是可以将整个图保存为protocol buffer。包括参数和操作。然而图还能被加载进其他支持的语言(C++、Java)
部署 yTorch的一个非常大的劣势就是没有办法很方便地部署模型,facebook和Microsoft一起搞了一个神经交换机,ONNX,可以将pytorch model转换到Caffe2 model,这样一是麻烦,二是Caffe2目前还在测试,一堆bug,用的人也不多,三是还要多学一个框架Caffe2 对于小规模的服务器端部署(例如一个Flask web server),两个框架都很简单。对于移动端和嵌入式部署,TensorFlow更好。不只是比PyTorch好,比大多数深度学习框架都要要。使用TensorFlow,部署在Android或iOS平台时只需要很小的工作量,至少不必用Java或者C++重写模型的推断部分
数据加载 PyTorch中用于加载数据的API设计的很棒。接口由一个数据集、一个取样器和一个数据加载器构成。数据加载器根据取样器的计划,基于数据集产生一个迭代器。并行化数据加载简单的就像把num_workers参数传递给数据加载器一样简单。 TensorFlow中没有发现特别有用的数据加载工具。很多时候,并不总能直接把准备并行运行的预处理代码加入TensorFlow图。以及API本身冗长难学
设备管理 PyTorch中,即使启用了CUDA,你也需要明确把一切移入设备。 TensorFlow的设备管理非常好用。通常你不需要进行调整,因为默认的设置就很好。例如,TensorFlow会假设你想运行在GPU上(如果有的话)
自定义扩展 PyTorch中,你只需为每个CPU和GPU编写一个接口和相应的实现。两个框架中编译扩展也是直接记性,并不需要在pip安装的内容之外下载任何头文件或者源代码。 两个框架都可以构建和绑定用C、C++、CUDA编写的自定义扩展。TensorFlow仍然需要更多的样板代码,尽管这对于支持多类型和设备可能更好

三、安装部署Pytorch
Pytorch官网给出了安装方式

先安装anaconda下载安装地址:
https://www.anaconda.com/distribution/#download-section
在这里插入图片描述
1.如果知道CUDA版本号?
liunx

查看 CUDA 版本:
cat /usr/local/cuda/version.txt
查看 CUDNN 版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

windows

win+R 或者cmd
nvcc --version
nvcc -v
或者 进入文件路径
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
如果没有找到需要下载cuda-toolkit

下载地址:https://developer.nvidia.com/cuda-toolkit
在这里插入图片描述
前提必须安装的python的环境、然后拷贝命令进行安装。

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

在这里插入图片描述

torchvision安装、输入cmd

conda install torchvision -c pytorch

报错:
“python setup.py egg_info” failed with error code 1 in C:\Users\nick\AppData\Local\Temp\pip-install-o_fy6mnh\torch\

pip3 install --no-deps torchvision

在这里插入图片描述
验证:

验证pytorch是否安装成功,在终端输入:
python
此时进入python环境。然后,继续输入
import torch
import torchvision

在这里插入图片描述

参考:https://zhuanlan.zhihu.com/p/28636490
参考:https://blog.csdn.net/sunqiande88/article/details/80085569

发布了226 篇原创文章 · 获赞 515 · 访问量 68万+

猜你喜欢

转载自blog.csdn.net/qq_32447301/article/details/88775163