初识PyTorch:从安装到入门,从入门到放弃

简介: 初识PyTorch:从安装到入门,从入门到放弃

PyTorch


PyTorch是Facebook团队于2017年1月发布的一个深度学习框架,虽然晚于TensorFlow,也没有TensorFlow火,但目前已经与TensorFlow奇虎相当。


而且PyTorch采用了Python语言的接口,可以说它才是Python程序员最容易上手的深度学习框架。


它就像GPU的Numpy,与Python一样都属于动态框架。PyTorch继承了Torch的灵活,动态的编程环境和友好的用户界面,支持以快速和灵活的方式构建动态神经网络。


还允许在训练的过程中,快速更改代码而不妨碍其性能,支持动态图形等尖端AI模型的能力,是快速实验的理想选择。


安装配置

安装PyTorch时,需要检查当前环境是否有GPU。如果没有GPU,那么就需要安装CPU版本,如果有GPU,则安装GPU版本。


(1)首先,我们需要安装NVIDIA驱动,安装网址如下:

https://www.nvidia.cn/Download/index.aspx?lang=cn

打开该网址,输入你的显卡参数,点击搜索即可下载。然后通过下一步下一步进行安装,安装成功之后,输入命令如下图所示:



(2)安装cuDNN


NVIDIA cuDNN是用于深度神经的GPU加速库。注册NVIDIA并下载cuDNN包,下载网址如下:

https://developer.nvidia.com/rdp/cudnn-archive


(3)安装Python以及PyTorch


通过一下网址选择自己的对应的版本,然后生成指定的命令:

https://pytorch.org/

复制后面的命令,然后安装即可,安装效果如下图所示:


安装验证

到这里,我们的GPU版PyTorch就安装成功了。当然,我们还需要验证是否安装成功,示例代码如下所示:

import torch
from torch.backends import cudnn
x = torch.tensor([10.0])
x = x.cuda()
print(x)
print(cudnn.is_acceptable(x))


运行之后,效果如下,即代表成功。

相关实践学习
基于阿里云DeepGPU实例,用AI画唯美国风少女
本实验基于阿里云DeepGPU实例,使用aiacctorch加速stable-diffusion-webui,用AI画唯美国风少女,可提升性能至高至原性能的2.6倍。
相关文章
|
5天前
|
机器学习/深度学习 算法 PyTorch
用PyTorch轻松实现二分类:逻辑回归入门
用PyTorch轻松实现二分类:逻辑回归入门
用PyTorch轻松实现二分类:逻辑回归入门
|
5天前
|
并行计算 PyTorch 算法框架/工具
anaconda+pytorch+opencv安装及环境配置
anaconda+pytorch+opencv安装及环境配置
159 0
|
8月前
|
机器学习/深度学习 PyTorch 算法框架/工具
Pytorch 最全入门介绍,Pytorch入门看这一篇就够了1
Pytorch 最全入门介绍,Pytorch入门看这一篇就够了
93 0
|
3天前
|
机器学习/深度学习 JSON PyTorch
图神经网络入门示例:使用PyTorch Geometric 进行节点分类
本文介绍了如何使用PyTorch处理同构图数据进行节点分类。首先,数据集来自Facebook Large Page-Page Network,包含22,470个页面,分为四类,具有不同大小的特征向量。为训练神经网络,需创建PyTorch Data对象,涉及读取CSV和JSON文件,处理不一致的特征向量大小并进行归一化。接着,加载边数据以构建图。通过`Data`对象创建同构图,之后数据被分为70%训练集和30%测试集。训练了两种模型:MLP和GCN。GCN在测试集上实现了80%的准确率,优于MLP的46%,展示了利用图信息的优势。
10 1
|
5天前
|
机器学习/深度学习 人工智能 并行计算
PyTorch入门指南:从安装到基础操作
【4月更文挑战第17天】PyTorch入门指南介绍了安装、Tensor操作、自动微分、神经网络构建及训练。安装PyTorch可通过官网选择相应环境的预构建包或使用conda命令。Tensor是基本计算单元,支持多种操作。自动微分简化了梯度计算。使用`torch.nn`定义神经网络,如示例中的全连接网络。训练网络涉及前向传播、损失计算、反向传播和权重更新,常用优化器如SGD。模型评估与测试包括在验证集或测试集上计算性能指标,如准确率。本文为初学者提供了PyTorch基础操作的快速入门。
|
5天前
|
机器学习/深度学习 人工智能 PyTorch
【Hello AI】安装和使用AIACC-AGSpeed(优化PyTorch深度学习模型)
AIACC-AGSpeed(简称AGSpeed)专注于优化PyTorch深度学习模型在阿里云GPU异构计算实例上的计算性能,相比原始的神龙AI加速引擎AIACC,可以实现无感的计算优化性能。本文为您介绍安装和使用AGSpeed的方法。
|
5天前
|
机器学习/深度学习 数据可视化 PyTorch
Pytorch 最全入门介绍,Pytorch入门看这一篇就够了(二)
Pytorch 最全入门介绍,Pytorch入门看这一篇就够了
67 2
|
5天前
|
机器学习/深度学习 PyTorch 算法框架/工具
Pytorch 最全入门介绍,Pytorch入门看这一篇就够了(一)
Pytorch 最全入门介绍,Pytorch入门看这一篇就够了
138 2
|
5天前
|
机器学习/深度学习 PyTorch 算法框架/工具
PyTorch 入门之旅
PyTorch 入门之旅
|
5天前
|
机器学习/深度学习 并行计算 PyTorch
win10安装minconda+pytorch
本文主要说明在win10笔记本上的安装minconda和pytorch,以图文结合的方式记录安装步骤,供参考学习。
64 1