pytorch判断是否cuda 判断变量类型方式


Posted in Python onJune 23, 2020

我就废话不多说了,那就直接看代码吧~

inputs = Variable(torch.randn(2,2))
inputs.is_cuda # will return false
inputs = Variable(torch.randn(2,2).cuda())
inputs.is_cuda # returns true

判断:

torch.is_tensor() #如果是pytorch的tensor类型返回true

torch.is_storage() # 如果是pytorch的storage类型返回ture

这里还有一个小技巧,如果需要判断tensor是否为空,可以如下

>>> a=torch.Tensor()
>>> len(a)
0
>>> len(a) is 0
True

设置:通过一些内置函数,可以实现对tensor的精度, 类型,print打印参数等进行设置

torch.set_default_dtype(d) #对torch.tensor() 设置默认的浮点类型
 
torch.set_default_tensor_type() # 同上,对torch.tensor()设置默认的tensor类型
>>> torch.tensor([1.2, 3]).dtype   # initial default for floating point is torch.float32
torch.float32
>>> torch.set_default_dtype(torch.float64)
>>> torch.tensor([1.2, 3]).dtype   # a new floating point tensor
torch.float64
>>> torch.set_default_tensor_type(torch.DoubleTensor)
>>> torch.tensor([1.2, 3]).dtype # a new floating point tensor
torch.float64
 
torch.get_default_dtype() #获得当前默认的浮点类型torch.dtype
 
torch.set_printoptions(precision=None, threshold=None, edgeitems=None, linewidth=None, profile=None)#)
## 设置printing的打印参数

判断变量类型:下面两种方法都行

if isinstance(downsample, torch.nn.Module):
# if torch.type(downsample) != torch.IntTensor:

补充知识:pytorch:测试GPU是否可用

废话不多说,看代码吧~

import torch
flag = torch.cuda.is_available()
print(flag)

ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())
True
cuda:0
GeForce GTX 1080
tensor([[0.9530, 0.4746, 0.9819],
  [0.7192, 0.9427, 0.6768],
  [0.8594, 0.9490, 0.6551]], device='cuda:0')

以上这篇pytorch判断是否cuda 判断变量类型方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
kNN算法python实现和简单数字识别的方法
Nov 18 Python
Python实现程序的单一实例用法分析
Jun 03 Python
python中的格式化输出用法总结
Jul 28 Python
python2.7 json 转换日期的处理的示例
Mar 07 Python
python 匹配url中是否存在IP地址的方法
Jun 04 Python
对python requests的content和text方法的区别详解
Oct 11 Python
Python面向对象程序设计多继承和多态用法示例
Apr 08 Python
django mysql数据库及图片上传接口详解
Jul 18 Python
用Python抢火车票的简单小程序实现解析
Aug 14 Python
详解用Python为直方图绘制拟合曲线的两种方法
Aug 21 Python
tesserocr与pytesseract模块的使用方法解析
Aug 30 Python
Idea安装python显示无SDK问题解决方案
Aug 12 Python
Pytorch 解决自定义子Module .cuda() tensor失败的问题
Jun 23 #Python
python如何查看安装了的模块
Jun 23 #Python
pytorch cuda上tensor的定义 以及减少cpu的操作详解
Jun 23 #Python
Python dict的常用方法示例代码
Jun 23 #Python
Python3 webservice接口测试代码详解
Jun 23 #Python
python和JavaScript哪个容易上手
Jun 23 #Python
pytorch 查看cuda 版本方式
Jun 23 #Python
You might like
PHP获取163、gmail、126等邮箱联系人地址【已测试2009.10.10】
2009/10/11 PHP
PHP网页游戏学习之Xnova(ogame)源码解读(十六)
2014/06/30 PHP
PHP图片自动裁切应付不同尺寸的显示
2014/10/16 PHP
thinkPHP实现基于ajax的评论回复功能
2018/06/22 PHP
JavaScript代码简单实现求杨辉三角给定行的最大值
2013/10/29 Javascript
jquery validate添加自定义验证规则(验证邮箱 邮政编码)
2013/12/04 Javascript
Javascript selection的兼容性写法介绍
2013/12/20 Javascript
js实现飞入星星特效代码
2014/10/17 Javascript
Bootstrap中data-target 到底是什么
2017/02/14 Javascript
JS中正则表达式全局匹配模式 /g用法详解
2017/04/01 Javascript
微信小程序商城项目之购物数量加减(3)
2017/04/17 Javascript
JS实现的Unicode编码转换操作示例
2017/04/28 Javascript
详解vue 组件之间使用eventbus传值
2017/10/25 Javascript
JS同步、异步、延迟加载的方法
2018/05/05 Javascript
详解Vue SSR( Vue2 + Koa2 + Webpack4)配置指南
2018/11/13 Javascript
JointJS流程图的绘制方法
2018/12/03 Javascript
微信小程序解析富文本过程详解
2019/07/13 Javascript
在vue中使用el-tab-pane v-show/v-if无效的解决
2020/08/03 Javascript
使用jquery实现轮播图效果
2021/01/02 jQuery
机器学习经典算法-logistic回归代码详解
2017/12/22 Python
详解Python用三种方式统计词频的方法
2019/07/29 Python
python 怎样将dataframe中的字符串日期转化为日期的方法
2019/09/26 Python
python实现梯度下降法
2020/03/24 Python
Python实现对word文档添加密码去除密码的示例代码
2020/12/29 Python
乌克兰最大的家用电器和电子产品连锁店:Eldorado
2019/10/02 全球购物
美国最大的购物网站:Amazon.com(亚马逊美国)
2020/05/23 全球购物
能源工程专业应届生求职信
2014/03/01 职场文书
大学学习计划书范文
2014/05/02 职场文书
优秀乡村医生事迹材料
2014/05/28 职场文书
优秀员工评优方案
2014/06/13 职场文书
个人遵守党的政治纪律情况对照检查材料思想汇报
2014/09/25 职场文书
交通事故委托书范本
2014/09/28 职场文书
转让协议书
2015/01/27 职场文书
2015年数学教师工作总结
2015/05/20 职场文书
pyqt5打包成exe可执行文件的方法
2021/05/14 Python
nginx rewrite功能使用场景分析
2022/05/30 Servers