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实现信用卡系统(支持购物、转账、存取钱)
Jun 24 Python
Python 编码处理-str与Unicode的区别
Sep 06 Python
python+django加载静态网页模板解析
Dec 12 Python
python机器学习之决策树分类详解
Dec 20 Python
让Django支持Sql Server作后端数据库的方法
May 29 Python
python提取具有某种特定字符串的行数据方法
Dec 11 Python
Python微医挂号网医生数据抓取
Jan 24 Python
Python 计算任意两向量之间的夹角方法
Jul 05 Python
Python pip使用超时问题解决方案
Aug 03 Python
Python collections.deque双边队列原理详解
Oct 05 Python
基于Python爬取素材网站音频文件
Oct 21 Python
opencv 分类白天与夜景视频的方法
Jun 05 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
如何开始收听短波广播
2021/03/01 无线电
PHP 简单数组排序实现代码
2009/08/05 PHP
PHP下利用shell后台运行PHP脚本,并获取该脚本的Process ID的代码
2011/09/19 PHP
PHP文件读取功能的应用实例
2015/05/08 PHP
自写的利用PDO对mysql数据库增删改查操作类
2018/02/19 PHP
PHP函数积累总结
2019/03/19 PHP
只需20行代码就可以写出CSS覆盖率测试脚本
2013/04/24 Javascript
javascript制作的cookie封装及使用指南
2015/01/02 Javascript
ECMAScript6块级作用域及新变量声明(let)
2015/06/12 Javascript
JS的框架Polymer中的dom-if和is属性使用说明
2015/07/29 Javascript
PHP+MySQL+jQuery随意拖动层并即时保存拖动位置实例讲解
2015/10/09 Javascript
再谈JavaScript异步编程
2016/01/27 Javascript
JavaScript实现弹出DIV层同时页面背景渐变成半透明效果
2016/03/25 Javascript
AngularJS中的指令全面解析(必看)
2016/05/20 Javascript
Angularjs根据json文件动态生成路由状态的实现方法
2017/04/17 Javascript
解决Nodejs全局安装模块后找不到命令的问题
2018/05/15 NodeJs
JS div匀速移动动画与变速移动动画代码实例
2019/03/26 Javascript
nodejs的安装使用与npm的介绍
2019/09/11 NodeJs
vue组件是如何解析及渲染的?
2021/01/13 Vue.js
Python开发WebService系列教程之REST,web.py,eurasia,Django
2014/06/30 Python
python使用正则表达式的search()函数实现指定位置搜索功能
2017/11/10 Python
浅谈Python中的作用域规则和闭包
2018/03/20 Python
浅谈Tensorflow模型的保存与恢复加载
2018/04/26 Python
python实现简单多人聊天室
2018/12/11 Python
pyqt5 comboBox获得下标、文本和事件选中函数的方法
2019/06/14 Python
Python序列化与反序列化pickle用法实例
2019/11/11 Python
Python pickle模块实现对象序列化
2019/11/22 Python
Tensorflow之梯度裁剪的实现示例
2020/03/08 Python
Python实现播放和录制声音的功能
2020/08/12 Python
CSS3实现同时执行倾斜和旋转的动画效果
2016/10/27 HTML / CSS
北美三大旅游网站之一:Travelocity
2017/08/12 全球购物
香港通票:Hong Kong Pass
2019/02/26 全球购物
新西兰最大、占有率最高的综合性药房:PharmacyDirect药房中文网
2020/11/03 全球购物
合作意向协议书
2015/01/29 职场文书
Ajax是什么?Ajax高级用法之Axios技术
2021/04/21 Javascript
MySQL快速插入一亿测试数据
2021/06/23 MySQL