如何判断pytorch是否支持GPU加速


Posted in Python onJune 01, 2021

加上这句代码:

print torch.cuda.is_available()

如何判断pytorch是否支持GPU加速

判断完毕!说说在pytorch中如何查看gpu信息吧~

为什么将数据转移至GPU的方法叫做.cuda而不是.gpu,就像将数据转移至CPU调用的方法是.cpu?这是因为GPU的编程接口采用CUDA,而目前并不是所有的GPU都支持CUDA,只有部分Nvidia的GPU才支持。

PyTorch未来可能会支持AMD的GPU,而AMD GPU的编程接口采用OpenCL,因此PyTorch还预留着.cl方法,用于以后支持AMD等的GPU。

torch.cuda.is_available()

cuda是否可用;

torch.cuda.device_count()

返回gpu数量;

torch.cuda.get_device_name(0)

返回gpu名字,设备索引默认从0开始;

torch.cuda.current_device()

返回当前设备索引;

如何判断pytorch是否支持GPU加速

以上为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python实现自主查询实时天气
Jun 22 Python
python实现一组典型数据格式转换
Dec 15 Python
[机器视觉]使用python自动识别验证码详解
May 16 Python
python线程的几种创建方式详解
Aug 29 Python
Django项目创建到启动详解(最全最详细)
Sep 07 Python
Python使用Opencv实现图像特征检测与匹配的方法
Oct 30 Python
python 画3维轨迹图并进行比较的实例
Dec 06 Python
python图形开发GUI库pyqt5的详细使用方法及各控件的属性与方法
Feb 14 Python
Python matplotlib修改默认字体的操作
Mar 05 Python
Python count函数使用方法实例解析
Mar 23 Python
使用keras实现BiLSTM+CNN+CRF文字标记NER
Jun 29 Python
Django中Q查询及Q()对象 F查询及F()对象用法
Jul 09 Python
pytorch 两个GPU同时训练的解决方案
Jun 01 #Python
使用Django实现商城验证码模块的方法
Jun 01 #Python
pytorch通过训练结果的复现设置随机种子
Jun 01 #Python
matplotlib画混淆矩阵与正确率曲线的实例代码
Jun 01 #Python
详细介绍python类及类的用法
教你怎么用PyCharm为同一服务器配置多个python解释器
用python批量解压带密码的压缩包
May 31 #Python
You might like
上海牌131型七灯四波段四喇叭一级收音机
2021/03/02 无线电
PHP 一个随机字符串生成代码
2010/05/26 PHP
php获取操作系统语言代码
2013/11/04 PHP
php获取新浪微博数据API实例
2013/11/12 PHP
Codeigniter通过SimpleXML将xml转换成对象的方法
2015/03/19 PHP
php+mysql实现的二级联动菜单效果详解
2016/05/10 PHP
Laravel5.7 Eloquent ORM快速入门详解
2019/04/12 PHP
利用PHP计算有多少小于当前数字的数字方法示例
2020/08/26 PHP
js操作select控件的几种方法
2010/06/02 Javascript
基于jquery的页面划词搜索JS
2010/09/14 Javascript
js实现双向链表互联网机顶盒实战应用实现
2011/10/28 Javascript
Extjs grid添加一个图片状态或者按钮的方法
2014/04/03 Javascript
js中string转int把String类型转化成int类型
2014/08/13 Javascript
JavaScript性能优化总结之加载与执行
2016/08/11 Javascript
JS写谷歌浏览器chrome的外挂实例
2018/01/11 Javascript
原生js调用json方法总结
2018/02/22 Javascript
vue配置nprogress实现页面顶部进度条
2019/09/21 Javascript
Bootstrap实现模态框效果
2019/09/30 Javascript
JavaScript 双向链表操作实例分析【创建、增加、查找、删除等】
2020/04/28 Javascript
JavaScript进阶(一)变量声明提升实例分析
2020/05/09 Javascript
js实现拖拽元素选择和删除
2020/08/25 Javascript
python开发之字符串string操作方法实例详解
2015/11/12 Python
Python快速从注释生成文档的方法
2016/12/26 Python
简单了解Python中的几种函数
2017/11/03 Python
Python 实现「食行生鲜」签到领积分功能
2018/09/26 Python
使用Python快速制作可视化报表的方法
2019/02/03 Python
pandas计算最大连续间隔的方法
2019/07/04 Python
Python可变对象与不可变对象原理解析
2020/02/25 Python
SmartBuyGlasses台湾:名牌眼镜,名牌太阳眼镜及隐形眼镜
2017/01/04 全球购物
印尼在线旅游门户网站:NusaTrip
2019/11/01 全球购物
个人教师自我评价范文
2013/12/02 职场文书
如何撰写岗位职责
2014/02/01 职场文书
2014年采购员工作总结
2014/11/18 职场文书
安全生产先进个人总结
2015/02/15 职场文书
工作服管理制度范本
2015/08/06 职场文书
Python 快速验证代理IP是否有效的方法实现
2021/07/15 Python