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脚本
Apr 05 Python
使用IPython下的Net-SNMP来管理类UNIX系统的教程
Apr 15 Python
python函数形参用法实例分析
Aug 04 Python
一步步教你用Python实现2048小游戏
Jan 19 Python
使用Python的turtle模块画图的方法
Nov 15 Python
python面向对象法实现图书管理系统
Apr 19 Python
Python tkinter模版代码实例
Feb 05 Python
Python如何把十进制数转换成ip地址
May 25 Python
Matlab中plot基本用法的具体使用
Jul 17 Python
Python基于traceback模块获取异常信息
Jul 23 Python
Python实现JS解密并爬取某音漫客网站
Oct 23 Python
python元组打包和解包过程详解
Aug 02 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入门速成(2)
2006/10/09 PHP
使用php shell命令合并图片的代码
2011/06/23 PHP
php截取中文字符串函数实例
2015/02/23 PHP
php去掉文件前几行的方法
2015/07/29 PHP
基于PHP的加载类操作以及其他两种魔术方法的应用实例
2017/08/28 PHP
JQuery的一些小应用收集
2010/03/27 Javascript
JS获取各种浏览器窗口大小的方法
2014/01/14 Javascript
手机平板等移动端适配跳转URL的js代码
2014/01/25 Javascript
js实现省市联动效果的简单实例
2014/02/10 Javascript
js给网页加上背景音乐及选择音效的方法
2015/03/03 Javascript
JavaScript获取并更改input标签name属性的方法
2015/07/02 Javascript
JavaScript中日期的相关操作方法总结
2015/10/24 Javascript
基于JavaScript实现表单密码的隐藏和显示出来
2016/03/02 Javascript
AngularJS基础 ng-dblclick 指令用法
2016/08/01 Javascript
bootstrap table小案例
2016/10/21 Javascript
bootstrap配合Masonry插件实现瀑布式布局
2017/01/18 Javascript
vue2.0构建单页应用最佳实战
2017/04/01 Javascript
利用ECharts.js画K线图的方法示例
2018/01/10 Javascript
Vant picker 多级联动操作
2020/11/02 Javascript
如何在vue中使用kindeditor富文本编辑器
2020/12/19 Vue.js
python实现ip查询示例
2014/03/26 Python
Python文本相似性计算之编辑距离详解
2016/11/28 Python
win10环境下python3.5安装步骤图文教程
2017/02/03 Python
浅谈python正则的常用方法 覆盖范围70%以上
2018/03/14 Python
Python/ArcPy遍历指定目录中的MDB文件方法
2018/10/27 Python
详解Python匿名函数(lambda函数)
2019/04/19 Python
Python对接支付宝支付自实现功能
2019/10/10 Python
HTML5手指下滑弹出负一屏阻止移动端浏览器内置下拉刷新功能的实现代码
2020/04/10 HTML / CSS
美国娱乐和流行文化商品店:FYE
2017/09/14 全球购物
师范学院毕业生求职信范文
2013/12/26 职场文书
八一演出活动方案
2014/02/03 职场文书
体育系毕业生求职自荐信
2014/04/16 职场文书
公安个人四风问题对照检查及整改措施
2014/10/28 职场文书
党支部季度考核意见
2015/06/02 职场文书
运动会主持人开幕词
2016/03/04 职场文书
面试分析分布式架构Redis热点key大Value解决方案
2022/03/13 Redis