深度学习--pytorch
文章平均质量分 71
深度学习框架pytorch的使用
p1049990866
这个作者很懒,什么都没留下…
展开
-
pytorch基础数据类型--tensor
文章目录一. 创建tensor二. tensor属性1. 判断属性is_cuda 是否存储在GPU上is_quantizedis_meta2. 基本属性dtype 数据类型device 计算引擎shape 形状3. 操作T 转置real 返回复数实部的新tensorimag 返回复数虚部的新tensor4. 其它grad 梯度ndim三. tensor方法1. 构造新tensornew_tensor 创建新tensornew_full 创建指定形状指定值的tensornew_empty 创建指定形状的空te原创 2021-04-10 20:20:50 · 1019 阅读 · 0 评论 -
安装pytorch
一. 普通安装pip install torch -i https://mirrors.aliyun.com/pypi/simple/验证是否安装成功import torchx = torch.rand(5, 3)print(x)二. GPU版本安装如果是nvidia显卡可在此处下载CUDA工具包并选择适合自己电脑的版本自行安装选择pytorch版本在官网选择对应自己cuda版本的pytorch进行安装# 示例pip install torch==1.8.1+cu111 torc原创 2021-04-10 12:06:17 · 429 阅读 · 0 评论