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爬取附近餐馆信息代码示例
Dec 09 Python
Python处理CSV与List的转换方法
Apr 19 Python
Python变量类型知识点总结
Feb 18 Python
如何利用Pyecharts可视化微信好友
Jul 04 Python
详解解决Python memory error的问题(四种解决方案)
Aug 08 Python
Django3.0 异步通信初体验(小结)
Dec 04 Python
pytorch 批次遍历数据集打印数据的例子
Dec 30 Python
tensorflow 利用expand_dims和squeeze扩展和压缩tensor维度方式
Feb 07 Python
python使用梯度下降算法实现一个多线性回归
Mar 24 Python
pandas DataFrame 数据选取,修改,切片的实现
Apr 24 Python
解决c++调用python中文乱码问题
Jul 29 Python
Python实现京东抢秒杀功能
Jan 25 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可应用于面包屑导航的迭代寻找家谱树实现方法
2015/02/02 PHP
Laravel框架实现的使用smtp发送邮件功能示例
2019/03/12 PHP
CL vs ForZe BO5 第二场 2.13
2021/03/10 DOTA
div层的移动及性能优化
2010/11/16 Javascript
初窥JQuery(一)jquery选择符 必备知识点
2010/11/25 Javascript
JavaScript自定义DateDiff函数(兼容所有浏览器)
2012/03/01 Javascript
JS链式调用的实现方法
2013/03/07 Javascript
js函数获取html中className所在的内容并去除标签
2013/09/08 Javascript
上传文件返回的json数据会被提示下载问题解决方案
2014/12/03 Javascript
JavaScript清空数组元素的两种方法简单比较
2015/07/10 Javascript
JS实现点击按钮控制Div变宽、增高及调整背景色的方法
2015/08/05 Javascript
Angular.JS学习之依赖注入$injector详析
2016/10/20 Javascript
canvas实现粒子时钟效果
2017/02/06 Javascript
Vue.js教程之axios与网络传输的学习实践
2017/04/29 Javascript
jquery实现提示语淡入效果
2017/05/05 jQuery
Js实现京东无延迟菜单效果实例(demo)
2017/06/02 Javascript
ReactNative Image组件使用详解
2017/08/07 Javascript
vue 中基于html5 drag drap的拖放效果案例分析
2018/11/01 Javascript
微信小程序实现判断是分享到群还是个人功能示例
2019/05/03 Javascript
Python使用SQLite和Excel操作进行数据分析
2018/01/20 Python
Python字典遍历操作实例小结
2019/03/05 Python
Python3 requests文件下载 期间显示文件信息和下载进度代码实例
2019/08/16 Python
Window系统下Python如何安装OpenCV库
2020/03/05 Python
Android Q之气泡弹窗的实现示例
2020/06/23 Python
使用TensorBoard进行超参数优化的实现
2020/07/06 Python
解决redis与Python交互取出来的是bytes类型的问题
2020/07/16 Python
python如何提升爬虫效率
2020/09/27 Python
基于canvas的骨骼动画的示例代码
2018/06/12 HTML / CSS
美国正宗奢华复古手袋、珠宝及配饰网站:What Goes Around Comes Around
2018/07/21 全球购物
Unix里面如何在后台运行程序
2016/10/14 面试题
信息技术专业大学生个人的自我评价
2013/10/05 职场文书
计算机专业自我鉴定
2013/10/15 职场文书
新农村建设典型材料
2014/05/31 职场文书
本科毕业生应聘自荐信范文
2014/06/26 职场文书
公司领导班子对照材料
2014/08/18 职场文书
乡镇八一建军节活动方案
2014/08/24 职场文书