如何判断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 网络编程起步(Socket发送消息)
Sep 06 Python
python使用pil生成图片验证码的方法
May 08 Python
Django中使用locals()函数的技巧
Jul 16 Python
Python简明入门教程
Aug 04 Python
Python 爬虫多线程详解及实例代码
Oct 08 Python
python下实现二叉堆以及堆排序的示例
Sep 29 Python
TensorFlow入门使用 tf.train.Saver()保存模型
Apr 24 Python
Pandas DataFrame 取一行数据会得到Series的方法
Nov 10 Python
Python3.5字符串常用操作实例详解
May 01 Python
解决python中用matplotlib画多幅图时出现图形部分重叠的问题
Jul 07 Python
python实现人工蜂群算法
Sep 18 Python
python在package下继续嵌套一个package
Apr 14 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
第十一节--重载
2006/11/16 PHP
Smarty模板学习笔记之Smarty简介
2014/05/20 PHP
php生成图片缩略图的方法
2015/04/07 PHP
JQuery实现鼠标滑过显示导航下拉列表
2013/09/12 Javascript
JS获取文本框,下拉框,单选框的值的简单实例
2014/02/26 Javascript
Javascript玩转继承(二)
2014/05/08 Javascript
js实现按Ctrl+Enter发送效果
2014/09/18 Javascript
JavaScript控制浏览器全屏及各种浏览器全屏模式的方法、属性和事件
2015/12/20 Javascript
JS实现回到页面顶部动画效果的简单实例
2016/05/24 Javascript
AngularJS中的Promise详细介绍及实例代码
2016/12/13 Javascript
利用javascript实现的三种图片放大镜效果实例(附源码)
2017/01/23 Javascript
vue2.0实现移动端的输入框实时检索更新列表功能
2018/05/08 Javascript
JavaScript数组,JSON对象实现动态添加、修改、删除功能示例
2018/05/26 Javascript
nodejs中request库使用HTTPS代理的方法
2019/04/30 NodeJs
vue 实现特定条件下绑定事件
2019/11/09 Javascript
JS pushlet XMLAdapter适配器用法案例解析
2020/10/16 Javascript
简单介绍Python中的decode()方法的使用
2015/05/18 Python
常用python编程模板汇总
2016/02/12 Python
python的Crypto模块实现AES加密实例代码
2018/01/22 Python
对Python的zip函数妙用,旋转矩阵详解
2018/12/13 Python
选择Python写网络爬虫的优势和理由
2019/07/07 Python
详解python变量与数据类型
2020/08/25 Python
Python 连接 MySQL 的几种方法
2020/09/09 Python
Python实现列表索引批量删除的5种方法
2020/11/16 Python
Python使用pyenv实现多环境管理
2021/02/05 Python
Html5新增标签有哪些
2017/04/13 HTML / CSS
乌克兰数字设备、配件和智能技术的连锁商店:KTC
2020/08/18 全球购物
全球最大运动品牌的男装、女装和童装官方库存商:A&A Sports
2021/01/17 全球购物
实习会计求职自荐信范文
2014/03/10 职场文书
2015年重阳节慰问信
2015/03/23 职场文书
2016年春节慰问信息
2015/03/25 职场文书
居住证明范文
2015/06/17 职场文书
关于观后感的作文
2015/06/18 职场文书
班主任班级管理心得体会
2016/01/07 职场文书
python 网络编程要点总结
2021/06/18 Python
试了下Golang实现try catch的方法
2021/07/01 Golang