当前位置：网站首页>pytorch框架学习(2)使用GPU训练

pytorch框架学习(2)使用GPU训练

2022-08-10 05:29:00 【Time.Xu】

使用GPU进行训练&常见的Tensor&Hub模块

需要将数据和模型传入到CUDA里进行运算。

在代码上：
与传统的不同的是我们需要指定一个设备，这个设备就是设定为cuda:0
需要把模型放置到我们设定的设备中(CUDA中) .to(设备)
需要把数据放置到我呢设定的设备中(CUDA中） .to(设备)

在这里插入图片描述

常见的Tensor有哪些形式

形式备注代码&注释
Scalar数值dim是0，标量
Vector向量dim是1，在深度学习中通常指特征，例如词向量，某一个维度的特征等
matrix矩阵dim是2，一行就是一个向量。可以认为每一行是每一个人，每一列是同一种特征。 矩阵能做乘法能做内积
n-dimensal tensor高维张量Aijk…

强大的Hub模块

Hub是为了方便开发者调用已有的代码,是一个model zoo
torch.hub.load()
学习一般不用这个

原网站

版权声明
本文为[Time.Xu]所创，转载请带上原文链接，感谢
https://xingshixu.blog.csdn.net/article/details/125961338

边栏推荐

猜你喜欢

随机推荐