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 相关文章推荐
Python导入txt数据到mysql的方法
Apr 08 Python
Python编程之属性和方法实例详解
May 19 Python
Python中列表元素转为数字的方法分析
Jun 14 Python
Python的Flask框架及Nginx实现静态文件访问限制功能
Jun 27 Python
python 3.5下xadmin的使用及修复源码bug
May 10 Python
Python修改文件往指定行插入内容的实例
Jan 30 Python
Python简单基础小程序的实例代码
Apr 28 Python
python里dict变成list实例方法
Jun 26 Python
PyCharm中代码字体大小调整方法
Jul 29 Python
python实现tail -f 功能
Jan 17 Python
Python3.7在anaconda里面使用IDLE编译器的步骤详解
Apr 29 Python
使用SimpleITK读取和保存NIfTI/DICOM文件实例
Jul 01 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 加密/解密函数 dencrypt(动态密文,带压缩功能,支持中文)
2009/01/30 PHP
php实现的简单中文验证码功能示例
2017/01/03 PHP
详谈phpAdmin修改密码后拒绝访问的问题
2017/04/03 PHP
PHP7新增函数
2021/03/09 PHP
javascript dom 操作详解 js加强
2009/07/13 Javascript
JavaScript this调用规则说明
2010/03/08 Javascript
基于jquery的一个图片hover的插件
2010/04/24 Javascript
js列举css中所有图标的实现代码
2011/07/04 Javascript
document.execCommand()的用法小结
2014/01/08 Javascript
javascript定义变量时带var与不带var的区别分析
2015/01/12 Javascript
BootStrap 下拉菜单点击之后不会出现下拉菜单(下拉菜单不弹出)的解决方案
2016/12/14 Javascript
jQuery实现判断控件是否显示的方法
2017/01/11 Javascript
原生JS实现《别踩白块》游戏(兼容IE)
2017/02/20 Javascript
bootstrap suggest搜索建议插件使用详解
2017/03/25 Javascript
react.js CMS 删除功能的实现方法
2017/04/17 Javascript
vue里面使用mui的弹出日期选择插件实例
2018/09/16 Javascript
Vue中的vue-resource示例详解
2018/11/02 Javascript
js实现超级玛丽小游戏
2020/03/18 Javascript
[46:14]VGJ.T vs Liquid 2018国际邀请赛小组赛BO2 第一场 8.19
2018/08/21 DOTA
详解Python爬虫的基本写法
2016/01/08 Python
django+js+ajax实现刷新页面的方法
2017/05/22 Python
Python重新加载模块的实现方法
2018/10/16 Python
Python判断一个文件夹内哪些文件是图片的实例
2018/12/07 Python
python爬虫获取百度首页内容教学
2018/12/23 Python
Python Multiprocessing多进程 使用tqdm显示进度条的实现
2019/08/13 Python
Python使用matplotlib 画矩形的三种方式分析
2019/10/31 Python
详解Django ORM引发的数据库N+1性能问题
2020/10/12 Python
浅谈Selenium 控制浏览器的常用方法
2020/12/04 Python
html5 Canvas画图教程(8)—canvas里画曲线之bezierCurveTo方法
2013/01/09 HTML / CSS
美国领先的户外服装与装备用品店:Moosejaw
2016/08/25 全球购物
美国女士泳装店:Swimsuits For All
2017/03/02 全球购物
同学聚会策划方案
2014/06/06 职场文书
教师节倡议书
2014/08/30 职场文书
群众路线教育实践活动剖析材料
2014/09/30 职场文书
女性健康知识讲座通知
2015/04/23 职场文书
2015年生产部工作总结范文
2015/05/25 职场文书