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系列之浅谈pickle模块封装和拆封数据对象的方法
May 23 Python
Python中的模块和包概念介绍
Apr 13 Python
python安装cx_Oracle模块常见问题与解决方法
Feb 21 Python
Python通过future处理并发问题
Oct 17 Python
Python3.6+Django2.0以上 xadmin站点的配置和使用教程图解
Jun 04 Python
python应用文件读取与登录注册功能
Sep 23 Python
Python进行统计建模
Aug 10 Python
python 深度学习中的4种激活函数
Sep 18 Python
Django-simple-captcha验证码包使用方法详解
Nov 28 Python
Pycharm 如何一键加引号的方法步骤
Feb 05 Python
使用Python快速打开一个百万行级别的超大Excel文件的方法
Mar 02 Python
Python实现老照片修复之上色小技巧
Oct 16 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实现的功能是显示8条基色色带
2006/10/09 PHP
php中常用编辑器推荐
2007/01/02 PHP
php 仿Comsenz安装效果代码打包提供下载
2010/05/09 PHP
如何使用php判断服务器是否是HTTPS连接
2013/07/05 PHP
PHP+百度AI OCR文字识别实现了图片的文字识别功能
2019/05/08 PHP
让IE8支持DOM 2(不用框架!)
2009/12/31 Javascript
jQuery动态显示和隐藏datagrid中的某一列的方法
2013/12/11 Javascript
jQuery学习笔记之创建DOM元素
2015/01/19 Javascript
Dojo获取下拉框的文本和值实例代码
2016/05/27 Javascript
深入浅析JavaScript中的3DES
2016/08/24 Javascript
详解ES6中的let命令
2020/04/05 Javascript
基于BootstrapValidator的Form表单验证(24)
2016/12/12 Javascript
JS闭包用法实例分析
2017/03/27 Javascript
Vue实现按钮旋转和移动位置的实例代码
2018/08/09 Javascript
nodejs require js文件入口,在package.json中指定默认入口main方法
2018/10/10 NodeJs
React父子组件间的传值的方法
2018/11/13 Javascript
JS获取本地地址及天气的方法实例小结
2019/05/10 Javascript
JS实现json数组排序操作实例分析
2019/10/28 Javascript
Python是编译运行的验证方法
2015/01/30 Python
我用Python抓取了7000 多本电子书案例详解
2019/03/25 Python
Python分析彩票记录并预测中奖号码过程详解
2019/07/09 Python
Python实现剪刀石头布小游戏(与电脑对战)
2019/12/31 Python
tensorboard实现同时显示训练曲线和测试曲线
2020/01/21 Python
Python语法垃圾回收机制原理解析
2020/03/25 Python
python列表的逆序遍历实现
2020/04/20 Python
Python变量格式化输出实现原理解析
2020/08/06 Python
改变 Python 中线程执行顺序的方法
2020/09/24 Python
jupyter notebook 写代码自动补全的实现
2020/11/02 Python
HTML5在canvas中绘制复杂形状附效果截图
2014/06/23 HTML / CSS
阿提哈德航空官方网站:Etihad Airways
2017/01/06 全球购物
小学教师的个人自我鉴定
2013/10/24 职场文书
粗加工管理制度
2014/02/04 职场文书
争做文明公民倡议书
2014/08/29 职场文书
毕业生就业推荐表自我评价
2015/03/02 职场文书
社区党建工作总结2015
2015/05/13 职场文书
2016新春团拜会致辞
2015/08/01 职场文书