DL框架之PyTorch:深度学习框架PyTorch的简介、安装、使用方法之详细攻略(二)

简介: DL框架之PyTorch:深度学习框架PyTorch的简介、安装、使用方法之详细攻略

2、PyTorch的三个结构层次

  • tensor:张量,多维数组,可在GPU上跑。
  • variable:可记住tensor在计算图中的位置,可知道当前variable和之前variable的关系。
  • module:神经网络的层次,如全连接层、卷积层。


PyTorch的安装

pip install torch

pip install torch==0.4.1.post2

T1方法

git clone https://github.com/pytorch/vision

cd vision

python setup.py install

T2方法

pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-win_amd64.whl

pip3 install torchvision

T3方法

pip install torch-0.4.0-cp36-cp36m-win_amd64.whl

pip install torchvision

image.png

1、20181114更新版本到 torch-0.4.1


pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp36-cp36m-win_amd64.whl

pip3 install torchvision


(1)、测试


import torch  

print(torch.__version__)          #输出版本信息

print(torch.cuda.is_available())  #查看是否支持cuda




2、20200108更新版本到 torch 1.3.1




官网:https://pytorch.org/




执行命令:


pip3 install torch===1.3.1 torchvision===0.4.2 -f https://download.pytorch.org/whl/torch_stable.html

或者

pip install torch-1.3.1-cp36-cp36m-win_amd64.whl


image.png


3、Anaconda内安装torch的whl文件更新版本到 torch 1.3.1


pip install D:\ProgramData\Anaconda3\envs\torch-1.3.1-cp36-cp36m-win_amd64.whl


image.png



4、Anaconda内安装torch的whl文件更新版本到 torch 1.0.0


pip install torch===1.0.0 torchvision===0.2.1 -f https://download.pytorch.org/whl/cu100/torch_stable.html


image.png




PyTorch的使用方法


相关文章

Pytorch之CNN:基于Pytorch框架实现经典卷积神经网络的算法(LeNet、AlexNet、VGG、NIN、GoogleNet、ResNet)——从代码认知CNN经典架构



0、Pytorch基础知识


1、PyTorch是一个提供两个高级功能的python包:  

具有强GPU加速度的张量计算(如numpy)

深层神经网络建立在基于磁带的自动调整系统上

可以重用您最喜爱的python软件包,如numpy,scipy和Cython,以便在需要时扩展PyTorch。  


2、PyTorch在细粒度级别是由以下组件组成的库:

torch   像NumPy这样的Tensor图书馆,拥有强大的GPU支持

torch.autograd   一种基于磁带的自动分类库,支持所有可区分的Tensor操作手电筒。pytorch的自动求导工具包在torch.autograd中。

torch.nn   一个神经网络库与autograd设计了最大的灵活性。pytorch神经网络构建很容易,主要使用的包是torch.nn这个包,可以尝试使用pytorch构建一个简单的二层神经网络结构。


torch.optim   一种与torch.nn一起使用的优化包,具有标准优化方法,如SGD,RMSProp,LBFGS,Adam等。

torch.multiprocessing   python多处理,但是具有魔法内存共享的手电筒传感器跨过程。适用于数据加载和hogwild培训。

torch.utils   DataLoader,Trainer等实用功能为方便起见 torch.legacy(.nn / .optim) 由于向后兼容性原因,已经从割炬移植的旧代码


1、Pytorch三行命令即可运算


image.png



2、Pytorch加载预训练模型


import torch

import torchvision

alexnet= torchvision.models.models.alexnet(pretrained=True)

vgg16 = torchvision.models.vgg16(pretrained=True)

resnet101 = torchvision.models.resnet101(pretrained=True)

resnet152 = torchvision.models.resnet152(pretrained=True)

 


相关文章
|
8天前
|
机器学习/深度学习 PyTorch 测试技术
PyTorch 深度学习(GPT 重译)(四)(1)
PyTorch 深度学习(GPT 重译)(四)
25 2
|
8天前
|
机器学习/深度学习 PyTorch API
PyTorch 深度学习(GPT 重译)(三)(4)
PyTorch 深度学习(GPT 重译)(三)
40 3
|
8天前
|
机器学习/深度学习 存储 PyTorch
PyTorch 深度学习(GPT 重译)(三)(3)
PyTorch 深度学习(GPT 重译)(三)
28 2
|
8天前
|
机器学习/深度学习 PyTorch 算法框架/工具
PyTorch 深度学习(GPT 重译)(二)(3)
PyTorch 深度学习(GPT 重译)(二)
26 1
|
8天前
|
机器学习/深度学习 算法 PyTorch
PyTorch 深度学习(GPT 重译)(二)(2)
PyTorch 深度学习(GPT 重译)(二)
34 2
|
8天前
|
机器学习/深度学习 PyTorch 算法框架/工具
PyTorch 深度学习(GPT 重译)(二)(1)
PyTorch 深度学习(GPT 重译)(二)
27 2
|
8天前
|
存储 PyTorch 算法框架/工具
PyTorch 深度学习(GPT 重译)(一)(3)
PyTorch 深度学习(GPT 重译)(一)
30 2
|
3天前
|
机器学习/深度学习 存储 边缘计算
深度学习在图像识别中的应用与挑战
【4月更文挑战第23天】 随着人工智能技术的飞速发展,深度学习作为其重要分支之一,在图像识别领域取得了显著的成果。本文将探讨深度学习在图像识别中的应用,分析其优势和面临的挑战,并展望未来的发展趋势。
|
1天前
|
机器学习/深度学习 运维 监控
深度学习在图像识别中的应用与挑战
【4月更文挑战第25天】 随着人工智能技术的飞速发展,深度学习已成为推动计算机视觉领域进步的核心动力。本文旨在探讨深度学习技术在图像识别任务中的创新应用及其面临的主要挑战。我们将首先回顾深度学习的基本原理和关键技术,随后通过实际案例分析其在图像分类、目标检测及语义分割等方面的应用效果。最后,文中将讨论当前技术瓶颈,如数据不平衡、模型泛化能力以及计算资源限制等,并提出可能的解决方案。
|
1天前
|
机器学习/深度学习 自动驾驶 算法
探索基于深度学习的图像识别在自动驾驶中的应用
【4月更文挑战第25天】 随着人工智能技术的飞速发展,深度学习已成为推动自动驾驶领域革新的核心力量。本文旨在深入探讨深度学习技术在图像识别领域的最新进展及其在自动驾驶系统中的实际应用。通过分析卷积神经网络(CNN)和递归神经网络(RNN)等先进算法,我们揭示了如何有效提升自动驾驶汽车在复杂环境中的视觉感知能力。文中还将讨论深度学习模型在处理实时数据时所面临的挑战,以及为解决这些挑战而采取的创新方法。